NYOJ20吝啬的国度广度搜索

原题地址：点击打开链接

#include<stdio.h>
#include<string.h>
#include<vector>
#include<queue>
using namespace std;
vector<int>g[100010];
int pre[100010],used[100010],n;
void bfs(int s)
{
	int i,u,v;
	queue<int>que;
	que.push(s);
	used[s]=1;
	pre[s]=-1;
	while(!que.empty())
	{
		u=que.front();
		que.pop();
		for(i=0;i<g[u].size();i++)
		{
			v=g[u][i];
			if(used[v]!=1)
			{
				used[v]=1;
				pre[v]=u;        //记录v前一个点
				que.push(v);
			}
		}
	}
}
int main()
{
	int i,t,a,b,s;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&s);
		for(i=1;i<=n;i++)
			g[i].clear();
		memset(used,0,sizeof(used));
		for(i=0;i<n-1;i++)
		{
			scanf("%d%d",&a,&b);
			g[a].push_back(b);
			g[b].push_back(a);
		}
		bfs(s);
		for(i=1;i<=n;i++)
			printf("%d ",pre[i]);
		printf("\n");
	}
	return 0;
}