nyist 20 吝啬的国度（dfs）

Posted 2020-09-14 风中的簌雨

吝啬的国度

题目描述：

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

题目大意：

　　　　　n个城市，n-1条无向边，问从s城市到其他城市的前一个城市的编号。

解题思路：

　　　　　利用深度优先搜索，从s点开始搜起。节点个数的数据过大所以不能用邻接矩阵存边

　　　　　我这里用的容器（vector）模拟邻接表的存储。

AC代码：

 1 #include <stdio.h>
 2 #include <string.h>
 3 #include <algorithm>
 4 #include <vector>
 5 
 6 using namespace std;
 7 
 8 int f[100010];    // 记录到达每个城市的前一个城市的节点
 9 vector< int> vec[100010];  // 用容器模拟邻接表存储边的信息
10 void dfs(int x)
11 {
12     int len = vec[x].size();   // 计算出和x城市相连的城市个数
13     for (int i = 0; i < len; i ++)
14     {
15         if (f[vec[x][i]])   // 不走重复的路
16             continue;
17         f[vec[x][i]] = x;
18         dfs(vec[x][i]);
19     }
20 }
21 int main ()
22 {
23     int t,n,s,i,j,a,b;
24     scanf("%d",&t);
25     while (t --)
26     {
27         memset(f,0,sizeof(f));
28         memset(vec,0,sizeof(vec));
29         scanf("%d%d",&n,&s);
30         for (i = 0; i < n-1; i ++)
31         {
32             scanf("%d%d",&a,&b);    // 无向边
33             vec[a].push_back(b);
34             vec[b].push_back(a);
35         }
36         f[s] = -1;    dfs(s);
37         printf("%d",f[1]);
38         for (i = 2; i <= n; i ++)
39             printf(" %d",f[i]);
40         printf("\n");
41     }
42     return 0;
43 }