题目描述是这样的,在一个吝啬的国度里有N个城市,这N个城市间只有N-1条路把这个N个城市连接起来。现在,Tom在第S号城市,他有张该国地图,他想知道如果自己要去参观第T号城市,必须经过的前一个城市是几号城市(假设你不走重复的路)。 下面是比较好的代码,用了递归。点击打开链接
#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include <vector>
#include<cstring>
using namespace std;
#define MAX 100003
vector<int> G[MAX]; //相当于二维数组
int pir[MAX];
int M, N, S;
void read(int n)
{
int a, b;
for(int i=1; i<n; i++)
{
scanf("%d %d",&a, &b);
G[a].push_back(b);
G[b].push_back(a);
}
}
int dfs(int v, int fa)
{
int k = G[v].size(); //对于V的每个相邻的子结点
for(int i=0; i<k; i++)
{
int u = G[v][i];
if(u!=fa) dfs(u,pir[u]=v); //如果当前结点不等于父节点,就递归的访问它的以U为结点的子树
}
}
int main()
{
scanf("%d",&M);
while(M--)
{
scanf("%d %d",&N, &S);
memset(pir, 0,sizeof(pir));
pir[S]=-1;
read(N);
dfs(S,-1);
for(int i=1; i<N; i++)
printf("%d ",pir[i]);
printf("%d\n",pir[N]);
for(int i=1; i<=N; i++)
if(G[i].size()!=0)
G[i].clear();
}
system("pause");
return 0;
}