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;    //迭代器遍历vector    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++