1562.微博转发 - BFS按层遍历 + 有向图

https://www.acwing.com/problem/content/1564/

题目：

第一行给出n（n个用户） l（需要考虑的层数）

接下来n行格式：m——第i名用户关注的总人数  mi——这位用户关注的用户编号

最后有q个询问，问qi用户最大可能关注量

如果 B 是 A 的关注者，C 是 B 的关注者，那么 A 的第一层关注者是 B，第二层关注者是 C

思路：

对于每一个用户i，建立一个有向图

bfs遍历l层，遍历过用户的做标记 ，统计l层内未被标记的用户个数，即最大转发量

/**道阻且长，行则将至*author:Roye_ack
*/
import java.util.*;
import java.io.*;
import java.math.*;class Main
{static int N=1010,M=100010;static int[] h=new int[N],ne=new int[M],e=new int[M];static int idx,n,l;public static void add(int a,int b){e[idx]=b;ne[idx]=h[a];h[a]=idx++;}public static int bfs(int x){int[] st=new int[N];Queue q=new LinkedList<>();q.offer(x);st[x]=1;int res=0;for(int k=0;k0){var t=q.poll();for(int i=h[t];i!=-1;i=ne[i]){int j=e[i];if(st[j]==0){st[j]=1;q.offer(j);res++;}   }}}return res;}public static void main(String[] args) throws IOException{PrintWriter wt=new PrintWriter(new BufferedWriter(new OutputStreamWriter(System.out)));n=rd.nextInt();l=rd.nextInt();Arrays.fill(h,-1);for(int i=1;i<=n;i++){int cnt=rd.nextInt();while(cnt-->0){int x=rd.nextInt();add(x,i);}}int q=rd.nextInt();while(q-->0){int t=rd.nextInt();wt.println(bfs(t));}wt.flush();}static class rd{static BufferedReader bf=new BufferedReader(new InputStreamReader(System.in));static StringTokenizer tk=new StringTokenizer("");static String nextLine() throws IOException{return bf.readLine();}static String next() throws IOException{while(!tk.hasMoreTokens()) tk=new StringTokenizer(bf.readLine());return tk.nextToken();}static int nextInt() throws IOException{return Integer.parseInt(next());}static double nextDouble() throws IOException{return Double.parseDouble(next());}static long nextLong() throws IOException{return Long.parseLong(next());}static BigInteger nextBig() throws IOException{BigInteger d=new BigInteger(rd.nextLine());return d;}}
}/*class PII
{int x,y;PII(int x,int y){this.x=x;this.y=y;}
}*/