Nyoj 20 悭吝的国度（dfs）

Nyoj 20 吝啬的国度（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<iostream>
#include<string.h>
#include<vector>
using namespace std;
int ans[100005];
vector<int>V[100005];

void dfs(int root)
{
//    for(vector<int>::iterator it=V[root].begin();it!=V[root].end();it++)
//    {
//        if(ans[V[root][*it]])
//            continue;
//        ans[V[root][*it]]=root;
//        dfs(V[root][*it]);
//    }
    for(int it=0;it<V[root].size();it++)
    {
        if(ans[V[root][it]])
            continue;
        ans[V[root][it]]=root;
        dfs(V[root][it]);
    }
    return ;
}

int main()
{
    int M,N,S,a,b;
    cin>>M;
    while(M--)
    {  memset(ans,0,sizeof(ans));

        cin>>N>>S;

        for(int i=0; i<N-1; i++)
        {
            cin>>a>>b;
            V[a].push_back(b);
            V[b].push_back(a);
        }

        ans[S]=-1;

        dfs(S);
        for(int j=1; j<=N; j++)
            cout<<ans[j]<<" ";
        cout<<endl;
       for(int k=1;k<=N;k++)
    V[k].clear();

    }
    return 0;
}