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);     } }
点赞 评论

相关推荐

09-23 14:45
贵州大学 财务
勇敢求职牛牛:怎么9.2佬人手一个中信证券实习
点赞 评论 收藏
分享
牛客网
牛客网在线编程
牛客网题解
牛客企业服务