1076 Forwards on Weibo PAT甲级 BFS

题目详情 - 1076 Forwards on Weibo (pintia.cn)

题目大意：每个用户可能被若干个用户关注，当某个用户发帖的时候，他的关注者可以选择转发该帖子，并且转发的帖子也可以被他自己的关注者转发，不过同一个用户最多只能转发一次，而且转发的层数是有限的。在给定所有用户的关注情况，以及最大转发层数的条件下，求给出的用户发帖后能得到的最大转发数。

思路分析：因为涉及到图旳遍历，然后又和图的层数有关，故可以想到使用BFS遍历，相比普通的BFS模板，增加了对层数的判断，我的做法是创建一个数组用于存储当前的结点的层数，则访问某个结点所连接的点，其所连接的点的层数就是该结点层数+1。然后就可以将当前层数小于l作为入队条件，求出最大转发数。

#include<iostream>
using namespace std;
#include<algorithm>
#include<cstring>
#include<queue>
const int N=100010;

int e[N],ne[N],idx=0,h[N];
int d[N];//存储点的层数
bool st[N];//入队判断
int n,l;

void add(int a,int b){
	e[idx]=b;
	ne[idx]=h[a];
	h[a]=idx++;
}

int bfs(int x){
	int res=0;
	d[x]=0;
	queue<int>q;
	q.push(x);
	st[x]=true;
	while(q.size()){
		int num=0;
		int u=q.front();
		q.pop();
		for(int i=h[u];i!=-1;i=ne[i]){
			int j=e[i];
			if(!st[j]&&d[u]<l){//当满足条件时入队
				st[j]=true;
				res++;
				q.push(j);
				d[j]=d[u]+1;//更新层数
			}
		}
	}
	return res;
}

int main(){
	memset(h,-1,sizeof h);
    scanf("%d%d",&n,&l);
    for(int i=1;i<=n;i++){
    	int m;
    	scanf("%d",&m);
    	for(int j=1;j<=m;j++){
    		int u;
			scanf("%d",&u);
			add(u,i);//建立被关注者指向关注者有向边
		}
	}
	
	int k;
	scanf("%d",&k);
	for(int i=0;i<k;i++){
		memset(st,0,sizeof st);
        memset(d,0,sizeof d);
		int x;
		scanf("%d",&x);
		int ans=bfs(x);
		printf("%d\n",ans);
	}
	return 0;
}