关注
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); } }
查看原帖
点赞 评论
相关推荐
牛客热帖
更多
正在热议
更多
# 你小心翼翼的闯过多大的祸? #
3290次浏览 66人参与
# 找不到实习会影响秋招吗 #
1399485次浏览 13634人参与
# 实习没事做是福还是祸? #
3494次浏览 58人参与
# 重来一次,你会对开始求职的自己说 #
838次浏览 19人参与
# 2025年终总结 #
132873次浏览 2272人参与
# 考研人,我有话说 #
156514次浏览 1211人参与
# 哪些公司笔/面试难度大? #
7040次浏览 32人参与
# 实习简历求拷打 #
23262次浏览 246人参与
# 你觉得现在还能进互联网吗? #
29909次浏览 201人参与
# 携程工作体验 #
18909次浏览 66人参与
# 大厂VS公务员你怎么选 #
69103次浏览 638人参与
# 扒一扒那些奇葩实习经历 #
140121次浏览 1149人参与
# 找不到好工作选择GAP真的丢人吗 #
93643次浏览 1007人参与
# 那些我实习了才知道的事 #
252988次浏览 1784人参与
# 非技术投递记录 #
672812次浏览 6820人参与
# 机械求职避坑tips #
81027次浏览 531人参与
# 投格力的你,拿到offer了吗? #
154737次浏览 829人参与
# 第一份工作能做外包吗? #
93990次浏览 599人参与
# 作业帮求职进展汇总 #
85283次浏览 559人参与
# 秋招遇到的奇葩面试题 #
101227次浏览 416人参与