nyoj 20 吝啬的国度

Posted 2020-07-02 非我非非我

吝啬的国度

时间限制：1000 ms  |  内存限制：65535 KB

难度：3

 

描述

在一个吝啬的国度里有N个城市，这N个城市间只有N-1条路把这个N个城市连接起来。现在，Tom在第S号城市，他有张该国地图，他想知道如果自己要去参观第T号城市，必须经过的前一个城市是几号城市（假设你不走重复的路）。

 

输入

第一行输入一个整数M表示测试数据共有M(1<=M<=5)组
每组测试数据的第一行输入一个正整数N(1<=N<=100000)和一个正整数S(1<=S<=100000)，N表示城市的总个数，S表示参观者所在城市的编号
随后的N-1行，每行有两个正整数a,b(1<=a,b<=N)，表示第a号城市和第b号城市之间有一条路连通。

输出

每组测试数据输N个正整数，其中，第i个数表示从S走到i号城市，必须要经过的上一个城市的编号。（其中i=S时，请输出-1）

样例输入

1
10 1
1 9
1 8
8 10
10 3
8 6
1 2
10 4
9 5
3 7

样例输出

-1 1 10 10 9 8 3 1 1 8

注意：a  b两个城市之间是无向连接的不是有向的

题解：用vector存图，然后用dfs遍历每个点，找到这个点的父节点，并存下来

#include<stdio.h>
#include<string.h>
#include<vector>
#include<queue>
#include<cstdio> 
#include<string>
#include<math.h>
#include<algorithm>
#define LL long long
#define PI atan(1.0)*4
#define DD double
#define MAX 200100
#define mod 100
#define dian 1.000000011
#define INF 0x3f3f3f
using namespace std;
int s[MAX];
vector<int>map[MAX];
void dfs(int beg)
{
	int i,j;
	for(i=0;i<map[beg].size();i++)
	{
		if(s[map[beg][i]]==0)
		{
			s[map[beg][i]]=beg;
			dfs(map[beg][i]);
		}
	}
}
int main()
{
	int n,m,j,i,sum,l,t,k;
	int a,b;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&m);
		memset(map,0,sizeof(map));
		memset(s,0,sizeof(s));
		for(i=1;i<n;i++)
		{
			scanf("%d%d",&a,&b);
			map[a].push_back(b);
			map[b].push_back(a);
		}
		dfs(m);
		s[m]=-1;
		for(i=1;i<n;i++)
		    printf("%d ",s[i]);
		printf("%d",s[n]);
		printf("\n");
	} 
	return 0;
}