本题为可达鸭D3第九课题目
题目描述
在一个吝啬的国度里有N个城市,这N个城市间只有N-1条路把这个N个城市连接起来。现在,Tom在第S号城市,他有张该国地图,他想知道如果自己要去参观第T号城市,必须经过的前一个城市是几号城市(假设你不走重复的路)。
输入描述:
第一行输入一个正整数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)
样例输入输出
输入:
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
题意简述
有N个点构成一棵树,求从节点S去每一个节点时,必须经过的前一个节点。
做题思路
使用vector存储数据,深搜找节点 x的上一个节点,把答案直接保存在ans数组中,一搜到底,不需要回溯,直接输出ans数组即可。
注意:i=S时输出-1
AC代码
#include<bits/stdc++.h>
using namespace std;
const int N=100010;
int ans[N];//存储答案
vector<int>a[N];//使用vector存储数据
void dfs(int x)
{
for(int i=0;i<a[x].size();i++)//遍历所有从城市x能够到达的城市
{
if(ans[a[x][i]]==0)//如果这个城市没走过
{
ans[a[x][i]]=x;//把这个城市的上一个城市设为城市x
dfs(a[x][i]);//从这个城市开始,继续搜
}
}
return;
}
int main()
{
int n,s,u,v;
scanf("%d%d",&n,&s);
for(int i=1;i<n;i++)
{
scanf("%d%d",&u,&v);//输入数据
a[u].push_back(v);
a[v].push_back(u);
}//存入vector
memset(ans,0,sizeof(ans));
ans[s]=-1;//i=S时,输出-1
dfs(s);//从现在的城市开始搜
for(int i=1;i<=n;i++)
{
printf("%d ",ans[i]);//输出答案数组
}
return 0;
}