吝啬的国度 【vector】+【DFS】

来源:互联网 发布:51自学网单片机 编辑:程序博客网 时间:2024/06/07 20: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)
样例输入
110 11 91 88 1010 38 61 210 49 53 7
样例输出
-1 1 10 10 9 8 3 1 1 8
思路 : dfs
代码
#include<stdio.h>#include<string.h>#include<algorithm>#include<math.h>#include<vector>  //  STL 容器 #define M 100000+10using namespace std;int n,s;vector<int>map[M];int pre[M];void dfs(int st){for(int i=0;i<map[st].size();i++){if(!pre[map[st][i]])  // 要是没有添加父辈 (其实这里本应该是用Visti【】数组来表明是否用过的,但是这个pre【】 数组也可以变相的表示这中情况){pre[map[st][i]]=st;dfs(map[st][i]);}}}int main(){int m;scanf("%d",&m);while(m--){memset(pre,0,sizeof(pre));memset(map,0,sizeof(map));scanf("%d%d",&n,&s);for(int i=0;i<n-1;i++){int a,b;scanf("%d%d",&a,&b);map[a].push_back(b);  //  用法 将两个数字连接 map[b].push_back(a);}dfs(s);pre[s]=-1;for(int i=1;i<=n;i++){printf("%d ",pre[i]);}putchar('\n');}return 0;}
0 0
原创粉丝点击