PAT甲级1076

PAT甲级1076

题目大意：给出1-n个人他们喜欢的博主（注意第i人是这些博主的追随者，意思是后面给出的几个博主发文可以通过该人进行转发，指向不要搞反了），给出层数L，即最多可以转发几次，之后给出几个博主，问他们的潜在粉丝是多少（只要能在l
次转发到的都算潜在粉丝）。
主要是图的BFS，每轮考虑一个人都把他指向的直接粉丝数收纳进队列，然后挨个取出再遍历，达到L轮为止。其中设置全局数组visit记录是否访问过该人（注意一定要初始化！！！博主写这个代码花了10来分钟，这里visit调试花了1个多小时…）

#include <iostream>
#include <vector>
#include <queue>
#define MAXN 1001
using namespace std;

int follow[MAXN][MAXN]={0};
int n,l;
void BFS(int a[],int b[],int m){
	for(int i=0;i<m;i++){
		queue<int> q;
		bool visit[n]={false};
		for(int c=1;c<=n;c++) visit[c]=false;	//一定要初始化！！！！！
		int target=a[i];
		visit[target]=true;
		for(int k=0;k<l;k++){
			if(q.size()==0&&k==0){	//刚开始还没数据进队时
				for(int j=1;j<=n;j++){
					if(follow[target][j]>0&&!visit[j]){
						q.push(j);
						visit[j]=true;
					}
				}
			}
			else{
				int total=q.size();
				while(total-->0){
					target=q.front();
					q.pop();
					for(int j=1;j<=n;j++)
						if(follow[target][j]>0&&!visit[j]){
							q.push(j);
							visit[j]=true;
						}
				}
			}
			b[i]+=q.size();
		}
	}
}
int main(){
	int tmp,sum;
	scanf("%d%d",&n,&l);
	for(int i=1;i<=n;i++){
		scanf("%d",&sum);
		for(int j=0;j<sum;j++){
			scanf("%d",&tmp);
			follow[tmp][i]=1;
		}
	}
	scanf("%d",&sum);
	int query[sum]={0},ans[sum]={0};
	for(int i=0;i<sum;i++){
		scanf("%d",&query[i]);
	}
	BFS(query,ans,sum);
	for(int i=0;i<sum;i++) printf("%d\n",ans[i]);
	system("pause");
	return 0;
}