深度搜索

NYOJ-20 吝啬的国度【深搜】

分类： 搜索 2011-08-08 19:47  1310人阅读  评论(1)  收藏  举报

测试

描述

在一个吝啬的国度里有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

思路：

简单的深搜，加边的时候要加入双向边。然后深搜每个点即可。

将无根树化为有根树，对每个节点进行遍历，记录从起点开始的父节点。

代码如下：

[cpp]  view plain copy

1.    
2. #include<iostream>  
3. #include<cstdio>  
4. #include<cstring>  
5. #include<string>  
6. #include<vector>  
7. #include<algorithm>  
8. using namespace std;  
9. int pre[100005];  
10. vector<int>v[100005];  
11.   
12. void DFS(int cur)  
13. {  
14.     for(int i = 0; i < v[cur].size(); ++i)  
15.     {  
16.         if(pre[v[cur][i]]) continue; //若存在父节点则继续遍历  
17.         pre[v[cur][i]] = cur; //相连节点的父节点为cur  
18.         DFS(v[cur][i]); //深搜到底，把一条路上父节点全部找出  
19.     }  
20. }  
21.   
22. int main()  
23. {  
24.     int ncase, num, cur, i, x, y;  
25.     scanf("%d", &ncase);  
26.     while(ncase--)  
27.     {  
28.         memset(v, 0, sizeof(v));  
29.         memset(pre, 0, sizeof(pre));  
30.         scanf("%d%d", &num, &cur);  
31.         pre[cur] = - 1; //起点没有父节点  
32.         for(i = 0; i < num - 1; ++i)  
33.         {  
34.             scanf("%d%d", &x, &y);  
35.             v[x].push_back(y); //x与y相连  
36.             v[y].push_back(x); //y与x也肯定相连  
37.         }  
38.         DFS(cur); //起点开始深搜  
39.         for(i = 1; i <= num; ++i)  
40.             printf("%d ", pre[i]); //每个节点的父节点都保存在pri数组，输出即可  
41.     }  
42.     return 0;  
43. }  
44.