1076 Forwards on Weibo BFS + DFS

题目大意

给出每个用户关注的人的id,和转发最多的层数,求一个id发了条微博最多会有多少个人转发。

思路解析

题目给出的是“关注表”,为了方便,需要转换成“被关注表”,即可以被哪些人转发。
本题考查带深度的图的遍历,dfs和bfs各有特色,下面给出两种解法,但是更推荐bfs。bfs先天具有这种统计层数的优势,dfs虽然也可以实现,只是需要注意,对于访问过的节点,有可能成为另一条路径的转发节点,而这个节点在第一次被遍历时,已经被置为了true,所以这一点是需要考虑的。比如:
示例
当最大层数为2,1发出微博。当转发路径为1 -> 2 -> 3,此时达到最大转发层,转发个数为2,回到栈顶,再从1->3时,因为3已经被访问过,所以递归结束,返回2。答案显然错误,所以单单判断visit是不够的,还需检查当前节点所在层数。以3节点为例,当路径为1 -> 2 -> 3时,3是第2层,而当第二次1 -> 3时,3是第1层,因为小于之前的第2层,所以允许3进栈。

示例代码

1. 深度优先示例

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int n, l, m, k,nmax = -1;
vector<vector<int>> fans;//粉丝表
vector<bool> visited;
vector<int> ranks(1005,7);
void dfs(int id,int count) {
	if (count > l) return;
	if (!visited[id])nmax++;
	visited[id] = true;
	ranks[id] = count;
	for (int i = 0; i < fans[id].size(); i++) {
		if (!visited[fans[id][i]] || count + 1 < ranks[fans[id][i]]) {//既要考虑visit,又要考虑当前层数
			dfs(fans[id][i],count+1);
		}
	}
}
int main() {
	scanf("%d %d", &n, &l);
	fans.resize(n + 1);
	visited.resize(n + 1);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &m);
		for (int j = 0; j < m; j++) {
			int temp;
			scanf("%d", &temp);
			fans[temp].push_back(i);
		}
	}
	scanf("%d", &k);
	for (int i = 0; i < k; i++) {
		int id;
		scanf("%d", &id);
		fill(visited.begin(), visited.end(), false);
		fill(ranks.begin(),ranks.end(),7);
		dfs(id, 0);
		printf("%d\n", nmax);
		nmax = -1;
	}
	return 0;
}

2. 广度优先示例

#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
int n, l, m, k, nmax = 0;
vector<vector<int>> fans;//粉丝表
void bfs(int id) {
	queue<int> que;
	que.push(id);
	vector<bool> visited(1005,false);
	vector<int> ranks(1005, 0);
	visited[id] = true;
	while (!que.empty()) {
		int temp = que.front();
		que.pop();
		for (int i = 0; i < fans[temp].size(); i++) {//新的一层入队
			if (ranks[temp] >= l) break;
			if (!visited[fans[temp][i]]) {
				ranks[fans[temp][i]] = ranks[temp] + 1;
				nmax++;
				visited[fans[temp][i]] = true;
				que.push(fans[temp][i]);
			}
		}
	}
}
int main() {
	scanf("%d %d", &n, &l);
	fans.resize(n + 1);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &m);
		for (int j = 0; j < m; j++) {
			int temp;
			scanf("%d", &temp);
			fans[temp].push_back(i);
		}
	}
	scanf("%d", &k);
	for (int i = 0; i < k; i++) {
		int id;
		scanf("%d", &id);
		bfs(id);
		printf("%d\n", nmax);
		nmax = 0;
	}
	return 0;
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值