题目大意
给出每个用户关注的人的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]]) {
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;
}