《蓝桥杯每日一题》bfs·AcWing1562. 微博转发

import java.util.*;
public class Main {
    static int N=100010;
    static int h[]=new int[N],e[]=new int[N],ne[]=new int[N];
    static boolean st[] = new boolean[N];
    static int n,l,idx;
    public static void main(String[] args)  {
        Scanner sc=new Scanner(System.in);
        n=sc.nextInt();   l=sc.nextInt();
        Arrays.fill(h,-1);
        //将每个人的关注者连向自己
        for(int i=1;i<=n;i++){
            int n1=sc.nextInt();
            while (n1-->0){
             int x=sc.nextInt();
             //建邻接表
             add(x,i);
            }
        }
        int k=sc.nextInt();
        while (k-->0){
            int t=sc.nextInt();
            System.out.println(bfs(t));
        }
    }
    private static Integer bfs(int x){
        Arrays.fill(st,false);
        Queue<Integer> q=new LinkedList<>();
        q.offer(x);
        st[x]=true;
        int res=0;
        //枚举每一层层数
          for(int i=0;i<l;i++){
              int size=q.size();
               /* 枚举每一层的每一个点 判断该点的子节点是否被枚举过
            如果没有则把该子节点加入队列中 当枚举到这一层的最后一个点时
            该层的点已被全部删除 此时队列里只有下一层的点了 while循环结束 继续for循环枚举下一层*/
              while (size-->0) {
                  int t = q.poll();
                  for(int j=h[t];j!=-1;j=ne[j]){
                      int tt=e[j];
                      if(st[tt]==false){
                          q.offer(tt);
                          st[tt]=true;
                          res++;
                      }
                  }
              }
          }
        return  res;
    }
    private static void add(int a, int b) {
        e[idx]=b;  ne[idx]=h[a];  h[a]=idx++;
    }
}