NYOJ 20 吝啬的国度

题目链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=20

题目详情

描述

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

解题思路

考虑到题目N点有N-1路线，为无向无环图，故而任一点到另一点的路径是唯一的，故直接由起点进行dfs，存储好父节点，记录好元素的访问过与否

代码(C++)

#include <bits/stdc++.h>
using namespace std;
const int M = 100000+10;
vector<int> C[M];    //存储与某点联通的点
int vis[M];          //设置遍历标记
int way[M];          //记录结果
void dfs(int s){
    vis[s] = 1;
    for(vector<int>::iterator it=C[s].begin();it != C[s].end();it++){
        if(!vis[*it]){
            way[*it] = s;
            dfs(*it);
        }
    }
}
void clearVector(int N){   //清空vector
    for(int i=1;i<=N;i++){
        C[i].clear();
    }
}
int main(){
    int M;
    cin>>M;
    while(M--){
        int N,S;
        cin>>N>>S;
        memset(vis,0,sizeof(vis));
        memset(way,0,sizeof(way));
        for(int i=0;i<N-1;i++){
            int a,b;
            cin>>a>>b;
            C[a].push_back(b);
            C[b].push_back(a);
        }
        way[S] = -1;           //按题目要求将起点置为-1
        dfs(S);
        for(int i=1;i<=N;i++){
            cout<<way[i] << " ";
        }
        clearVector(N);        //清空vector,避免影响下一组数据
    }
    return 0;
}

运行结果

Result	Accept
Time	492MS
Memory	4244
Language	G++