关注
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); } }
查看原帖
点赞 评论
相关推荐
牛客热帖
更多
正在热议
更多
# 在大厂上班是一种什么样的体验 #
11125次浏览 139人参与
# 你的mentor是什么样的人? #
50862次浏览 716人参与
# 程序员找工作至少要刷多少题? #
19315次浏览 249人参与
# 机械人避雷的岗位/公司 #
43634次浏览 300人参与
# 我现在比当时_,你想录用我吗 #
8904次浏览 111人参与
# 我和mentor的爱恨情仇 #
106012次浏览 942人参与
# 设计人如何选offer #
189529次浏览 868人参与
# 为了减少AI幻觉,你注入过哪些设定? #
4969次浏览 151人参与
# 你的秋招进行到哪一步了 #
2529980次浏览 23253人参与
# 机械人还在等华为开奖吗? #
311987次浏览 1582人参与
# 牛客AI体验站 #
7000次浏览 189人参与
# 重来一次,我还会选择这个专业吗 #
411082次浏览 3898人参与
# 12306一秒售罄,你抢到回家的票了吗? #
2052次浏览 48人参与
# 论秋招对个人心气的改变 #
11529次浏览 165人参与
# 一张图晒一下你的AI员工 #
5261次浏览 117人参与
# AI Coding的使用心得 #
4798次浏览 101人参与
# 刚入职的你踩过哪些坑 #
7071次浏览 129人参与
# 关于春招/暑期实习,你想知道哪些信息? #
7727次浏览 120人参与
# 应届生进小公司有什么影响吗 #
118349次浏览 1159人参与
# 晒晒你司的新年福利 #
8641次浏览 107人参与
查看11道真题和解析