关注
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); } }
查看原帖
点赞 评论
相关推荐
迷茫的大四🐶:真的遇见佬了

点赞 评论 收藏
分享

点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 26届秋招公司红黑榜 #
7452次浏览 24人参与
# 实习必须要去大厂吗? #
145423次浏览 1531人参与
# 平安产险科技校招 #
2172次浏览 0人参与
# 校招泡的最久的公司是哪家? #
2863次浏览 15人参与
# 度小满求职进展汇总 #
9115次浏览 49人参与
# 你的mentor是什么样的人? #
1933次浏览 12人参与
# 智慧芽求职进展汇总 #
14043次浏览 96人参与
# 职场新人体验 #
93479次浏览 638人参与
# 你觉得mentor喜欢什么样的实习生 #
7744次浏览 237人参与
# 没有家庭托举的我是怎么找工作的 #
9567次浏览 148人参与
# 入职第一天,你准备什么时候下班 #
84975次浏览 466人参与
# 从哪些方向判断这个offer值不值得去? #
4904次浏览 83人参与
# 技术岗笔试题求解 #
95051次浏览 1101人参与
# 求职低谷期你是怎么度过的 #
3953次浏览 78人参与
# 最难的技术面是哪家公司? #
54426次浏览 893人参与
# 面试紧张时你会有什么表现? #
1029次浏览 19人参与
# 秋招想进国企该如何准备 #
97301次浏览 487人参与
# 机械人的工作环境真的很差吗 #
24527次浏览 119人参与
# 你有哪些缓解焦虑的方法? #
36846次浏览 835人参与
# 独居后,你的生活是更好了还是更差了? #
27771次浏览 263人参与
# 跳槽时有那些注意事项 #
105751次浏览 567人参与
# 工作压力大怎么缓解 #
117355次浏览 1108人参与