[DFS]吝啬的国度

描述

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

【采用深度优先搜索】

#include <stdio.h>
#include <string.h>
int map[100005];
void DFS(int start)
{
    int pre = map[start];
    if(pre!=0)
   {
      DFS(pre);
      map[pre]=start;
   }
}
int main()
{
    int m, n, start, a, b,i;
    scanf("%d",&m);
   while (m--)
   {
     scanf("%d%d",&n,&start);;
      memset(map, 0, sizeof(map)); //地图上所有元素清零
      for(i= 1; i < n; i++)
      {
          scanf("%d%d",&a,&b);
         if(map[b]==0)
                    map[b]=a;
         else
              {
                 DFS(a);
                  map[a]=b;
              }
     }
        DFS(start);
       map[start] = -1;
      for(i = 1; i < n; i++)
          printf("%d ",map[i]);
 printf("%d\n",map[n]);
}

return 0;
}

 ================== 优秀代码

#include <iostream>
#include <stdio.h>
#include <vector>
using namespace std;
vector<int>a[100005];
int b[100005],n,s;
void dfs(int x,int y){
    for (int i = 0; i <a[x].size();i++)
        if (a[x][i] != y)
            dfs(a[x][i],b[a[x][i]]=x);
        return;
}
int main(){
    int M,x,y;
    scanf("%d",&M);
    while (M--){
        scanf("%d%d",&n,&s);
        for (int i = 1; i <n; i++){
            a[i].clear();
        }
        for (int i = 1; i <n; i++){
            scanf("%d%d",&x,&y);
            a[x].push_back(y);         //相当二维数组 
            a[y].push_back(x);
        }
        b[s] = -1;               //与S相等则为-1  
        dfs(s,-1);
        for (int i = 1; i <= n; i++)
            printf("%d ",b[i]);
        printf("\n");
    }
    return 0;
}

深度优先详解：http://rapheal.iteye.com/blog/1526863