1、题目信息
吝啬的国度
时间限制:
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
2、思路
每个城市可以多有个相邻的城市,所以我们选择使用vector类,来构建邻接表,使用深度优先搜索法遍历城市图
3、代码
#include <stdio.h>
#include <vector>
#include <string.h>
#define maxSize 100010
using namespace std;
int pre[maxSize];//存储所有城市的前以个城市
vector<int> v[maxSize];//这里的v作为二维数组使用,优点是第二维长度可变
//深度优先搜索
void dfs(int s){
for(int i=0;i<v[s].size();i++){
if(!pre[v[s][i]]){//如果s城市邻接的城市的前驱城市没有标记
pre[v[s][i]]=s;
dfs(v[s][i]);//递归下一个城市节点
}
}
}
int main(){
int m,n,s,a,b;
scanf("%d",&m);
while(m--){
memset(pre,0,sizeof(pre));//对数组中每一个元素赋相同的值
memset(v,0,sizeof(v));
scanf("%d%d",&n,&s);
pre[s]=-1;
for(int i=1;i<n;i++){
scanf("%d%d",&a,&b);
v[a].push_back(b);//在当前数组中的末尾存入元素
v[b].push_back(a);
}
dfs(s);//从s节点深度遍历二维数组
for(int i=1;i<=n;i++){
printf("%d",pre[i]);
if(i!=n) printf(" ");
else printf("\n");
}
}
return 0;
}