nyoj 20 吝啬的国度 dfs+建立树

吝啬的国度

时间限制： 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

思路:我发现对于这样的需要自己建立树或者图的题目对我来说真的是个难点；

本题的要求需要我们建立的是一个无向图(一条边可以连通两个城市互相到达),最后,由于需要从s城市为起点到达t城市且需要记录他的前一个城市

是多少,我们用pre[i]来保存从s到i号城市所要经过的上一个城市,pre[a]=b,表示b指向a,那么问题来了a只能由一个城市指向,

否则它的值会发生丢失

初始化 pre[i]都为0；

  对于每组边的两节点 a b

  若 pre[a]=0，则没有城市指向,pre[a]=b；

  若pre[a]！=0，说明已经有城市指向a,我们就需要调整有向图的方向,即要让pre[b]=a,(去让a指向b)

但是让a指向b指向我们需要判断是否有城市指向了b，即pre[b]是否为0，若有则此值将会被覆盖,所以我们在将a指向b之前需要将指向b的箭头调

转方向，即如果t指向 b,则调转方向另b指向t（一个城市可以指向多个城市，但一个城市只能被一个城市指向）,依次递归下去，直到找到同一个

未被指向的,然后改变方向,从而我们就实现了有向图的180翻转;

  完成上述操作后,因为我们是要从s城市出发,即以s城市为树的根节点建立有向图,所以在对s城市进行上述递归就可以得到一个从s城市出发的有向

图;然后输出结果即可；

ac代码：

#include<stdio.h>
#include<string.h>
int n,m,s,pre[100010];
void dfs(int x)
{    int t=pre[x];
    if(t)
    {   dfs(t);
       pre[t]=x;
	}
}
int main()
{   
    scanf("%d",&m);
    while(m--)
    {   int a,b;
	   memset(pre,0,sizeof(pre));
	   scanf("%d %d",&n,&s);
       for(int i=1;i<n;i++)
        {  scanf("%d %d",&a,&b);
          if(pre[a]==0)
            pre[a]=b;
          else
          {  dfs(b);
             pre[b]=a;
		  }
		}
		dfs(s);
		pre[s]=-1;
		printf("%d",pre[1]);
		for(int i=2;i<=n;i++)
		 printf(" %d",pre[i]);
		 printf("\n");
		 
	}
	return 0;
}

第二种方法 vector建树加dfs

#include<stdio.h>
#include<string.h>
#include<vector>
#define N 100010
using namespace std;
vector<int>v[N];
int pre[N];
void dfs(int x){
	int i;
	for(i=0;i<v[x].size();i++){
		if(pre[v[x][i]])// 这里很重要;; 否则就是一直在递归第一个点;
		  continue;
		pre[v[x][i]]=x;
		dfs(v[x][i]);
	}
	return ;
}
int main(){
	int m,n,s,a,b;
	scanf("%d",&m);
	while(m--){
		memset(pre,0,sizeof(pre));
		memset(v,0,sizeof(v));
		scanf("%d%d",&n,&s);
		for(int i=1;i<=n-1;i++)
		{ scanf("%d%d",&a,&b);
		  v[a].push_back(b);
		  v[b].push_back(a);
		}
		pre[s]=-1;
		dfs(s); 
		printf("%d",pre[1]);
		for(int i=2;i<=n;i++)
		 printf(" %d",pre[i]);
		 printf("\n");
	}
	return 0;
}