题意:微博的关注链可以形成一张有向网,求其中某个人在L层内,所有关注的孩子节点;
思路:此题就是求所给定的某一个节点,求其在L层类所有孩子节点的个数,利用BFS广度搜索,如果当前父节点层次<L,把他的孩子节点加入队列,求出所有孩子节点个数。其孩子节点的个数包含L层的节点而不含其给定的根节点
代码:
#include<iostream>
#include<queue>
using namespace std;
struct node {
int data, level;
};
int n = 0, l = 0,num = 0,quire = 0,id= 0;
vector<vector<int>> v;
int BFS(int root) {
vector<int> p;
int cnt = 0;
bool vis[1010] = { false };
queue<node> q;
q.push(node{root,0});
vis[root] = true;
while (!q.empty()) {
node tem = q.front();
q.pop();
for (int i = 0; i <v[tem.data].size(); i++) {
if (vis[v[tem.data][i]] == false && tem.level < l) {//如果当前的节点层次小于l,把其未访问的孩子节点压入队列,总的人数使是要算上L层的
cnt++;
vis[v[tem.data][i]] = true;
p.push_back(v[tem.data][i]);
q.push(node{ v[tem.data][i],tem.level + 1 });
}
}
}
return cnt;
}
int main() {
cin >> n >> l;
v.resize(n+1);
for (int i = 1; i <= n; i++) {
scanf("%d",&num);
int tem = 0;
for (int j = 0; j < num; j++) {
scanf("%d",&tem);
v[tem].push_back(i);
}
}
scanf("%d",&quire);
for (int i = 0; i < quire; i++) {
scanf("%d",&id);
printf("%d\n", BFS(id));
}
system("pause");
return 0;
}