参观城市（dfs,改编spfa）

问题 F: 参观城市

时间限制: 1 Sec   内存限制: 128 MB
提交: 34   解决: 19
[ 提交][ 状态][ 讨论版]

题目描述

有N个城市，这N个城市间只有N-1条路把这个N个城市连接起来。现在，小明在第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）

样例输入

110 101 21 91 83 78 68 109 510 310 4

样例输出

8 1 10 10 9 8 3 10 1 -1

     竞赛的时候就差一点写出来！

我是想用邻接表做，我用了两种方法，一种是spfa的模板改编的，另一种是dfs

上代码！

dfs版本:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int per[110000],vis[110000],t,head[110000];
struct node 
{
	int u,v,next;
}s[110000];
void addedge(int x,int y)
{
	s[t].u=x;
	s[t].v=y;
	s[t].next=head[x];
	head[x]=t++;
}
void dfs(int x)
{
	for(int k=head[x];k!=-1;k=s[k].next)
	{
		int v=s[k].v;
		if(!vis[v])
		{
			per[v]=x;
			vis[v]=1;
			dfs(v);
		}
	}
}
int main()
{
	int N,a,b,i,m,n;
		scanf("%d",&N);
		while(N--)
		{
			memset(head,-1,sizeof(head));
			memset(vis,0,sizeof(vis));
			scanf("%d%d",&m,&n);
			t=0;
			for(i=0;i<n-1;i++)
			{
				scanf("%d%d",&a,&b);
				addedge(a,b);
				addedge(b,a);
			}
			vis[m]=1;
			per[m]=-1;
			dfs(m);
			for(i=1;i<m;i++)
			printf("%d ",per[i]);
			printf("%d\n",per[m]);
		}
return 0;
}

spfa改编版本：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
#define INF 0x3f3f3f3f
using namespace std;
int per[101000],vis[101000];
int head[101000];
int m,n;
struct node 
{
	int from,to,next;
}s[101000];
void spfa()
{
	int cn,i;
	memset(vis,0,sizeof(vis));
	queue<int> qq;
	vis[n]=1;
	cn=n;
	per[n]=-1;
	qq.push(n);
	while(!qq.empty())
	{
		int u=qq.front();
		qq.pop();
		cn=u;
		for(int k=head[u];k!=-1;k=s[k].next)
		{
			int v=s[k].to;
				if(!vis[v])
				{
					qq.push(v);
					per[v]=cn;
					vis[v]=1;
				}
		}
	}
	for(i=1;i<m;i++)
	printf("%d ",per[i]);
	printf("%d\n",per[m]);
}
int main()
{
	int i,j,k,l,t,x,y,a,b,N;
	scanf("%d",&N);
	while(N--)
	{
		scanf("%d%d",&m,&n);
		 t=0;
		 memset(head,-1,sizeof(head));
		for(i=0;i<m-1;i++)
		{
			scanf("%d%d",&a,&b);
			s[t].from=a;
			s[t].to=b;
			s[t].next=head[a];
			head[a]=t++;
			s[t].from=b;
			s[t].to=a;
			s[t].next=head[b];
			head[b]=t++;
		}
		spfa();
	}
	return 0;
}

两种方法的区别是，dfs是一搜到底，然后回溯！

而spfs版本是把和当前相连的搜索完，在进行下一个！

都很巧妙，然而vector我还不会用，下次一定补上！