PAT 1076 Forwards on Weibo

PAT 1076 Forwards on Weibo

BFS广搜，解决，DFS也可以解决，但是更麻烦，有时间补充解方案

#include <bits/stdc++.h>
using namespace std;

const int maxv=1010;

struct node
{
	int id;
	int layer;
};

vector<node> Adj[maxv];
bool inq[maxv]={false};

int BFS(int s, int L)
{
	int num=0;
	queue<node> q;
	node start;
	start.id=s;
	start.layer=0;
	q.push(start);
	inq[start.id]=true;
	while(!q.empty())
	{
		node top=q.front();
		q.pop();
		int u=top.id;
		for(int i=0;i<Adj[u].size();i++)
		{
			node next=Adj[u][i];
			next.layer=top.layer+1;
			if(inq[next.id]==false && next.layer<=L)
			{
				q.push(next);
				inq[next.id]=true;
				num++;
			}
			
		}
	}
	return num;
} 

int main()
{
	node user;
	int n,L,num,idF;
	cin>>n>>L;
	for(int i=1; i<=n; i++)
	{
		user.id=i;
		cin>>num;
		for(int j=0;j<num;j++)
		{
			cin>>idF; 
			Adj[idF].push_back(user);
		}
	}
	int numQuery,s;
	cin>>numQuery;
	for(int i=0;i<numQuery;i++)
	{
		memset(inq,false,sizeof(inq));
		cin>>s;
		int num=BFS(s,L);
		cout<<num<<endl; 
	}
	
	return 0;	
}