nyoj-20-吝啬的国度(深搜)

吝啬的国度

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

解题思路：

      一开始用了队列来求，起点入队，然后出队，找到跟起点相连的城市进队，那么这些城市的前一步就是出队的城市了，然后开始不停地出队入队，直到队空为止。思路不错，但是超时了。

      经过看大神的代码，发现这个题可以用深搜来解决，再一看题目发现这确实符合深搜的特性。

     不管是队列还是深搜，都需要标记。

     队列用到了#include<queue>STL函数，

#include <iostream>
#include <queue>
using namespace std; //这几个头文件必不可少

int main()
{
queue<类型（如int）> q; //使用前需定义一个queue变量，且定义时已经初始化
while(!q.empty()) q.pop(); //重复使用时，用这个初始化（空则返回1，不空返回0）
q.push(1); //进队列
q.pop(); //出队列
int v=q.front(); //得到队首的值
int s=q.size(); //得到队列里元素个数

return 0;
}

      因为给定的城市N的数目太大，建立数组需要用到#include<vector>，vector就是一个不定长数组，vector<int>a就是一个类似于int a[]的整数数组，只不过他的长度不确定，可以用a.size()读取他的长度。

      而vector<int>a[max]就是一个二维数组，只是第一维的大小是固定的（不超过max），二维的大小就不固定了，这道题之所以用到vector就是利用了他的不定长，如果直接建立二维数组a[n][n]，n太大了，这样的二维数组绝对超出内存。

      (1)头文件#include<vector>.

      (2)创建vector对象，vector<int> vec;

      (3)尾部插入数字：vec.push_back(a);

      (4)使用下标访问元素，cout<<vec[0]<<endl;记住下标是从0开始的。

      (5)向量大小:vec.size();

代码：

//队列思路（超时） 
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<queue>
using namespace std;
int sta[110000];
int map[110000][3];
int ans[110000];
int main()
{
	int m;
	int n,s;
	int i,j,k;
	int now;
	queue<int>q;
	scanf("%d",&m);
	while(m--)
	{
		scanf("%d%d",&n,&s);
		q.push(s);
		for(i=1;i<n;i++)
		{
			map[i][0]=1;
		    scanf("%d%d",&map[i][1],&map[i][2]);
		}
		memset(ans,0,sizeof(ans));
		ans[s]=-1;
		while(!q.empty())
		{
			now=q.front();
			q.pop();
			for(i=1;i<n;i++)
			{
				if(map[i][0]&&map[i][1]==now)
				{
				    q.push(map[i][2]);
				    ans[map[i][2]]=now;
				    map[i][0]=0;
				}
				else if(map[i][0]&&map[i][2]==now)
				{
				    q.push(map[i][1]);
				    ans[map[i][1]]=now;
				    map[i][0]=0;
				}
			}
		}
		for(i=1;i<=n;i++)
		{
			printf("%d",ans[i]);
			if(i!=n)
			    printf(" ");
			else
			    printf("\n");
		}
	}
	return 0;
}

//深搜代码（AC） 
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int pre[100005];
vector<int>v[100005];

void dfs(int s)
{
	int i;
	for(i=0;i<v[s].size();i++)
	{
		if(pre[v[s][i]])
		    continue;
		pre[v[s][i]]=s;
		dfs(v[s][i]);
	}
}
int main()
{
	int m;
	int n,s;
	int a,b;
	int i,j;
	scanf("%d",&m);
	while(m--)
	{
		memset(v,0,sizeof(v));
		memset(pre,0,sizeof(pre));
		scanf("%d%d",&n,&s);
		pre[s]=-1;
		for(i=0;i<n-1;i++)
		{
			scanf("%d%d",&a,&b);
			v[a].push_back(b);
			v[b].push_back(a);
		}
		dfs(s);
		for(i=1;i<=n;i++)
		{
		    printf("%d",pre[i]);
		    if(i!=n)
		        printf(" ");
		    else
		        printf("\n");
		}
	}
	return 0;
}