图的遍历问题,本题用BFS方法更为简便
坑点如下
1,本题要求返回层数小于等于l的间接转发数,则根结点层数为0
2,数组赋初值的方法
memset(vis,false,sizeof(vis));
3,BFS方法细节
int BFS(int first){
int num=0;
queue<int>Q;
Q.push(first);
node[first].level=0;
vis[first]=true;
while(!Q.empty()){
int q=Q.front();
Q.pop();
for(int i=0;i<node[q].follow.size();i++){
if(vis[node[q].follow[i]]==false&&node[q].level+1<=l)
{ node[node[q].follow[i]].level=node[q].level+1;
Q.push(node[q].follow[i]);
vis[node[q].follow[i]]=true;
num++;
}
}
}
return num;
}
本题中应保证每个结点只入队一次,入队后赋值结点层数,之后层数不再改变。如先赋值层数再入队,则在同层数结点赋值完毕后,在一层可能会改变本层兄弟结点的层数。
4,对队列处理时记得写Q.pop()否则会超时。
整体代码如下
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
using namespace std;
const int maxn=1010;
struct Node{
int level;
vector<int>follow;
}node[maxn];
bool vis[maxn]={false};
int n,l,num1,user;
int BFS(int first){
int num=0;
queue<int>Q;
Q.push(first);
node[first].level=0;
vis[first]=true;
while(!Q.empty()){
int q=Q.front();
Q.pop();
for(int i=0;i<node[q].follow.size();i++){
if(vis[node[q].follow[i]]==false&&node[q].level+1<=l)
{ node[node[q].follow[i]].level=node[q].level+1;
Q.push(node[q].follow[i]);
vis[node[q].follow[i]]=true;
num++;
}
}
}
return num;
}
int main(){
scanf("%d %d",&n,&l);
for(int i=1;i<=n;i++){
scanf("%d",&num1);
for(int l=0;l<num1;l++){
scanf("%d",&user);
node[user].follow.push_back(i);
}
}
int qnum,first;
scanf("%d",&qnum);
for(int i=0;i<qnum;i++){
memset(vis,false,sizeof(vis));
scanf("%d",&first);
printf("%d\n",BFS(first));
}
}