1076 Forwards on Weibo (30 分)

广度搜索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;
}