关注
package test2; import java.util.ArrayList; import java.util.HashMap; import java.util.HashSet; import java.util.LinkedList; import java.util.Queue; import java.util.Scanner; import javax.management.Query; public class Main3 { public static void main(String[] args) { Scanner scn = new Scanner(System.in); String[] str1 = scn.nextLine().split(" "); String[] str2 = scn.nextLine().split(" "); String ss = scn.nextLine(); int k = Integer.parseInt(ss); int[] arr1 = new int[str1.length]; int[] arr2 = new int[str2.length]; int tag=0; for(int i=0;i<arr1.length;i++) { arr1[i] = Integer.parseInt(str1[i]); arr2[i] = Integer.parseInt(str2[i]); if(k==arr1[i]||k==arr2[i]) tag = 1; } HashMap<Integer, ArrayList<Integer>> map = new HashMap<>(); for(int i=0;i<arr2.length;i++) { if(map.containsKey(arr2[i])) { ArrayList<Integer> list = map.get(arr2[i]); list.add(arr1[i]); map.replace(arr2[i], list); }else { ArrayList<Integer> list = new ArrayList<Integer>(); list.add(arr1[i]); map.put(arr2[i], list); } } HashSet<Integer> set = new HashSet<Integer>(); Queue<Integer> queue = new LinkedList<Integer>(); queue.offer(k); set.add(k); if(tag==0) { System.out.println(0); return; } int res = 1; while(!queue.isEmpty()) { Integer a = queue.poll(); // System.out.println(a); ArrayList<Integer> list = map.get(a); if(list!=null) { for(Integer b:list) { if(!set.contains(b)) { set.add(b); queue.offer(b); res++; } } } } System.out.println(res); } }
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 发面经攒人品 #
2711580次浏览 36745人参与
# 你觉得mentor喜欢什么样的实习生 #
3502次浏览 117人参与
# 智慧芽求职进展汇总 #
7812次浏览 24人参与
# 26届秋招公司红黑榜 #
3051次浏览 8人参与
# 没有家庭托举的我是怎么找工作的 #
5190次浏览 80人参与
# 从哪些方向判断这个offer值不值得去? #
2563次浏览 57人参与
# 平安产险科技校招 #
1957次浏览 0人参与
# 求职低谷期你是怎么度过的 #
2264次浏览 52人参与
# 我对___祛魅了 #
134487次浏览 743人参与
# 我的求职进度条 #
135859次浏览 1558人参与
# 你现在会用到哪些AI技能? #
12844次浏览 108人参与
# 实习必须要去大厂吗? #
144518次浏览 1518人参与
# 度小满求职进展汇总 #
7697次浏览 41人参与
# 入职第一天,你准备什么时候下班 #
84173次浏览 462人参与
# 你有哪些缓解焦虑的方法? #
36487次浏览 832人参与
# 牛客十周岁生日快乐 #
177975次浏览 1810人参与
# 一汽大众工作体验 #
11954次浏览 23人参与
# 互联网公司爆料 #
146527次浏览 710人参与
# 九号公司求职进展汇总 #
2117次浏览 18人参与
# 大厂VS公务员你怎么选 #
48494次浏览 538人参与
# 来聊聊机械薪资天花板是哪家 #
146259次浏览 804人参与
# 机械人的工作环境真的很差吗 #
23975次浏览 118人参与