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("
");
		}
	}
	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("
");
		}
	}
	return 0;
}



原文地址:https://www.cnblogs.com/brucemengbm/p/6892334.html