关注
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); } }
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# mt对你说过最有启发的一句话 #
1363次浏览 28人参与
# 秋招特别不鸣谢 #
1510次浏览 29人参与
# 被上班搭子“传染”了哪些习惯 #
997次浏览 37人参与
# 今年秋招你收到了多少封邮件? #
1625次浏览 42人参与
# 工作后,你落下了哪些病根 #
2491次浏览 77人参与
# 工作后明白的那些道理 #
33162次浏览 417人参与
# 选完offer后,你后悔学机械吗? #
49211次浏览 270人参与
# 巨人网络求职进展汇总 #
181358次浏览 1211人参与
# 你见过最离谱的招聘要求是什么? #
246054次浏览 1695人参与
# 当发现同事想辞职 #
12319次浏览 39人参与
# 选实习,你更看重哪方面? #
1555次浏览 42人参与
# 摸鱼被leader发现了怎么办 #
75573次浏览 429人参与
# 工作中遇到的歹人 #
4061次浏览 85人参与
# 上班到公司第一件事做什么? #
114728次浏览 799人参与
# 工作两年想退休了 #
209054次浏览 1855人参与
# 国企vs私企,怎么选? #
39446次浏览 214人参与
# 机械制造面试记录 #
295624次浏览 3129人参与
# 业务面应该做哪些准备 #
80856次浏览 835人参与
# 选offer应该考虑哪些因素 #
127971次浏览 947人参与
# 软开人,你觉得应届生多少薪资才算合理? #
95241次浏览 509人参与