题目大意:
在微博中,每个用户都可能被其他若干用户关注。每当该用户发布一条信息时,他的关注者就可以看到这条信息并选择是否转发它,且转发的信息也可以被转发者的关注者再次转发,但同一用户最多只转发该信息一次(信息的最初发布者不能转发该信息)。现在给出N个用户的关注情况(即他们各自关注了那些用户)以及一个转发层数上限L,并给出最初发送消息的因户编号,求在转发层数上限内消息最多会被多少用户转发——摘自《算法笔记.上机训练实战指南》
解题思路:
BFS层次遍历,为每个用户维护一个记录层次的变量即可,代码如下:
#include<iostream>
#include<cstdio>
#include<fstream>
#include<set>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<iomanip>
#include<cstdlib>
#include<list>
#include<queue>
#include<stack>
#include<algorithm>
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
//vector ::iterator it;
//set<int>::iterator iter;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int n,l,k,ma[1001][1010];
struct node
{
int level,data;//层次,数据
};
int bfs(int root,int level)
{
int num=0;//计数
node test;
test.data=root;test.level=level;
queue<node> q;
q.push(test);
int book[1010]={0};
book[root]=1;//标记首先发微博的人
while(!q.empty())
{
node tmp=q.front();
if(l==tmp.level) break;//到达最后一层,退出遍历
q.pop();
for(int i=1;i<=n;i++)
{
if(ma[tmp.data][i]&&!book[i])//如果是粉丝且没有访问过
{
book[i]=1;
num++;
node tmp1;
tmp1.data=i;tmp1.level=tmp.level+1;
q.push(tmp1);
}
}
}
return num;
}
int main()
{
// std::ios::sync_with_stdio(false);
// cin.tie(0);
// freopen("test.txt","r",stdin);
// freopen("output.txt","w",stdout);
scanf("%d %d",&n,&l);
for(int i=1;i<=n;i++)
{
int m;
scanf("%d",&m);
for(int j=0;j<m;j++)
{
int id;
scanf("%d",&id);
ma[id][i]=1;
}
}
scanf("%d",&k);
for(int i=0;i<k;i++)
{
int id;
scanf("%d",&id);
int num=bfs(id,0);
printf("%d\n",num);
}
return 0;
}