吝啬的国度
时间限制:1000 ms | 内存限制:65535 KB
难度:3
- 描述
- 在一个吝啬的国度里有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 此题应该用dfs或bfs
dfs代码如下#include <cstdio>
#include <vector>
#include <cstring> using namespace std; vector<int> path[];
int pre[]; int N, S; void dfs(int start) {
int cnt = path[start].size();
for(int i = ; i < cnt;i++) {
int p = path[start][i];
if(pre[p] == ) {
pre[p] = start;
dfs(p);
} }
}
int main(int argc, char const *argv[])
{
int M;
//freopen("input.txt","r",stdin);
scanf("%d",&M);
while(M--) {
memset(path, , sizeof(path));
memset(pre, , sizeof(pre));
scanf("%d %d",&N,&S);
for(int i = ; i < N-; i++) {
int x,y;
scanf("%d %d",&x,&y);
path[x].push_back(y);
path[y].push_back(x);
}
pre[S] = -;
dfs(S);
printf("%d", pre[]);
for(int i = ; i <= N; i++) {
printf(" %d",pre[i]);
}
puts("");
}
return ;
}bfs代码如下
#include <cstdio>
#include <vector>
#include <cstring>
#include <queue>
using namespace std; vector<int> path[];
int pre[];
queue <int> que;
int N, S; int main(int argc, char const *argv[])
{
int M;
//freopen("input.txt","r",stdin);
scanf("%d",&M);
while(M--) {
memset(path, , sizeof(path));
memset(pre, , sizeof(pre));
scanf("%d %d",&N,&S);
for(int i = ; i < N-; i++) {
int x,y;
scanf("%d %d",&x,&y);
path[x].push_back(y);
path[y].push_back(x);
}
pre[S] = -;
que.push(S); while(!que.empty()) {
int p0 = que.front();que.pop();
int cnt = path[p0].size();
for(int i = ; i < cnt;i++) {
int p = path[p0][i];
if(pre[p] == ) {
pre[p] = p0;
que.push(p);
}
}
}
printf("%d", pre[]);
for(int i = ; i <= N; i++) {
printf(" %d",pre[i]);
}
puts("");
}
return ;
}