题目链接:
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)
样例输入
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 |
算法思想:
这是一道典型的深度搜索的题目,这些城市与道路构成了一个极小连通子图,也就是生成树。由于节点数太大,不能使用二维数组存储,可以考虑使用vectora[100005]来存储,这样可以减少部分存储空间,但仍然是指标不治本。看了网上的一段代码,觉得可以借鉴,所以来过来记录一下。
首先,给定的是一个无向图,我们需要做的是将这个无向图构造存储为一个以开始城市节点为根节点的一棵树(有向图)。这样就可以知道每个城市的父节点,然后循环输出即可。
构造这颗树的方法:首先,使用mp[i]来记录从S节点到i节点必须经过的上一个城市节点的编号。
比如:mp[b] = a,也就是在有向图中a指向b,即要达到b城市必须经过的上一个城市是a。
初始化 mp[i]=0;
则对于每组边的两节点a,b
若mp[b]=0,即无城市指向b,则令a指向b:mp[b]=a
若mp[b]!=0, 即表明已经有其他城市指向b,那么,这时当然要将b指向a城市,即mp[a]=b;
但是在将b指向a之前,需要考虑是已经有其他城市已经指向了a,如果已经有的话,冒然赋值则会覆盖以前的值。
所以,在此之前,我们需要将所有指向a的城市节点,方向调换,即若t指向a,则改为a指向t
同时,再以t为起始递归下去,这样就可以让已a为起点的一系列有向线路全部转换方向,实现180转弯。
经过上述操作我们已经得到一个有向图,但仍然不能满足题目要求,因为起点是S,故每个有向线路的起点必然是由S开始的,
所以,我们对S点,进行上述的递归操作,此时,任务完成,所有的有向路都是已s点为出发点的了。
源代码
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include <cstring>
int m, n, s, mp[100005];
void def(int s)
{
int t = mp[s];
if (t != 0)
{
def(t);
mp[t] = s;
}
}
int main(void)
{
scanf("%d", &m);
while (m--)
{
int a, b;
scanf("%d%d", &n, &s);
memset(mp, 0, sizeof(int)*n + 1);
for (int i = 1; i<n; i++)
{
scanf("%d%d", &a, &b);
if (mp[a] == 0)
{
mp[a] = b;
}
else
{
def(b);
mp[b] = a;
}
}
def(s);
mp[s] = -1;
for (int i = 1; i <= n; i++)
{
printf("%d ", mp[i]);
}
printf("\n");
}
return 0;
}