广度搜索BFS
注意理解题意,关注者与被关注者的关系。
同时L是表示,在L层以内的所有节点,包括L层。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e4+5;
vector<int> G[maxn];
int vis[maxn];
int N,L,cnt;
struct node{
int value,layer;
};
void bfs(int r){
queue<node> q;
node root;
root.layer = 0;
root.value = r;
q.push(root);
vis[r] = 1;
while(!q.empty()){
node tmp = q.front();
q.pop();
int u = tmp.value;
for(int i=0;i<G[u].size();i++){
int v = G[u][i];
root.layer = tmp.layer + 1;
root.value = v;
if(vis[v]==0 && root.layer <= L){
q.push(root);
vis[v]=1;
cnt++;
}
}
}
}
int main(){
int u,v,k,x;
scanf("%d%d",&N,&L);
for(int j=1;j<=N;j++){
scanf("%d",&k);
for(int i=0;i<k;i++){
scanf("%d",&x);
G[x].push_back(j);
}
}
scanf("%d",&k);
for(int i=0;i<k;i++){
memset(vis,0,sizeof(vis));
scanf("%d",&x);
cnt = 0;
bfs(x);
printf("%d\n",cnt);
}
return 0;
}
深度搜索DFS
最后一个测试点超时
27分
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e4+5;
vector<int> G[maxn];
int vis[maxn];
int N,L,cnt;
struct node{
int value,layer;
};
void dfs(int r,int depth){
if(depth > L) return;
if(vis[r]==0){
cnt++;
vis[r]=1;
}
for(int i=0;i<G[r].size();i++){
int v = G[r][i];
dfs(v, depth+1);
}
}
int main(){
int u,v,k,x;
scanf("%d%d",&N,&L);
for(int j=1;j<=N;j++){
scanf("%d",&k);
for(int i=0;i<k;i++){
scanf("%d",&x);
G[x].push_back(j);
}
}
scanf("%d",&k);
for(int i=0;i<k;i++){
memset(vis,0,sizeof(vis));
scanf("%d",&x);
cnt = 0;
dfs(x,0);
printf("%d\n",cnt-1);
}
return 0;
}