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

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值