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;
}