题目详情 - 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;
}