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