1076. Forwards on Weibo (30)

最新推荐文章于 2023-05-29 15:45:16 发布

原创最新推荐文章于 2023-05-29 15:45:16 发布 · 287 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#C++ #PAT #Forwards on Weibo

PAT(Advanced Level) 专栏收录该内容

103 篇文章

订阅专栏

本文介绍了一种使用广度优先搜索(BFS)算法来解决社交网络中影响力传播问题的方法。具体而言，通过建立一个关注关系图，并利用BFS算法来计算从特定用户出发在固定层级内的转发数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

考虑BFS，需要注意的就是A follow B 形成的路径是 B->A，因为 A 关注 B 则 B发的微博可以被 A 转发，然后每次 query 的时候走一遍 BFS 查找。

#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>

using namespace std;

vector<vector<bool>> graph;

int main(){
	int n, l, k;
	scanf("%d%d", &n, &l);
	graph.resize(n+1, vector<bool>(n+1, false));

	for(int i = 1; i <= n; ++i){
		int num, id;
		scanf("%d", &num);
		for(int j = 0; j < num; ++j){
			scanf("%d", &id);
			graph[id][i] = true;
		}
	}

	scanf("%d", &k);
	for(int i = 0; i < k; ++i){
		int id;
		scanf("%d", &id);

		queue<pair<int,int>> Q;
		Q.emplace(id, 0);

		vector<bool> used(n+1, false);
		used[id] = true;

		int sum = 0;
		// bfs
		while(!Q.empty()){
			auto p = Q.front(); Q.pop();
			int u = p.first, level = p.second;
			
			if(level == l) break;

			for(int i = 1; i <= n; ++i){
				if(!used[i] && graph[u][i]){
					used[i] = true;
					++sum;
					Q.emplace(i, level+1);
				}
			}
		}

		printf("%d\n", sum);
	}

	return 0;
}