吝啬的国度
时间限制:
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
-
-
分析:
由于只能从S出发,且不走重复的路,
-
此题可以转化为求二叉树各节点的父亲节点 ,刚开始Tom在第S号城市,S即总父亲,对于样例数据的二叉树如下:
所对应的2-10号的父亲分别为1,10,10,9,8,3,1,1,8.这里用vector容器保存父子节点。
-
第一行输入一个整数M表示测试数据共有M(1<=M<=5)组
v[s][i] 表示父亲 s 有 i 个儿子,分别为v[s][0],v[s][i]......v[s][i - 1].
采用深搜从始点依次遍历整棵树,保存父亲结点。
代码如下:
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<map>
#include<vector>
using namespace std;
const int maxn = 1e5 + 5;
int ans[maxn]; //ans[i] = j : 表示参观第 i 号城市,必须经过的前一个城市是第 j 号城市
vector<int> v[maxn];
void dfs(int s)
{
for(int i = 0; i <v[s].size(); i ++) //遍历s的儿子
{
if(ans[v[s][i]] != 0) //若已经找到父亲
continue;
ans[v[s][i]] = s; //s的儿子的父亲就是s,ans[s的儿子] = s
dfs(v[s][i]); //将s的儿子作为父亲,继续搜索
}
}
int main()
{
int T, N, S, a, b;
scanf("%d", &T);
while(T--)
{
scanf("%d%d", &N, &S);
memset(ans, 0, sizeof(ans));
memset(v, 0, sizeof(v));
for(int i = 1; i < N; i ++)
{
scanf("%d%d", &a, &b);
v[a].push_back(b); //a的父亲是b,b的父亲是a,但根据题意只有一个对应关系成立,如上图
v[b].push_back(a);
}
ans[S] = -1;
dfs(S);
for(int i = 1; i <= N; i ++)
printf("%d ", ans[i]);
}
return 0;
}