1713: 参观城市 （最短路变形 记录城市号 好题）

1713: 参观城市

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

样例输入

1

10 10

 1 2

1 9

1 8

3 7

8 6

8 10

9 5

10 3

10 4

样例输出

8 1 10 10 9 8 3 10 1 -1

 

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
#define MIN(x,y)(x<y?x:y)
using namespace std;
const int INF=0x3f3f3f3f;
const int MAXN=100010;
const int MAXM=200010;
int top,N,S;
struct Edge
{
	int from,to,val,next;
};
Edge edg[MAXM];
int head[MAXM],dis[MAXN],vis[MAXN];
void add(int u,int v,int w)
{
	Edge E={u,v,w,head[u]};
	edg[top]=E;
	head[u]=top++;
}
int SPFA(int now)
{
	queue<int>q;
	memset(vis,0,sizeof(vis));
	memset(dis,INF,sizeof(dis));
	q.push(S);
	dis[S]=0;
	vis[S]=1;
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		vis[u]=0;
		for(int i=head[u];i!=-1;i=edg[i].next)
		{
			int v=edg[i].to;
			if(dis[v]>dis[u]+edg[i].val)
			{
				dis[v]=dis[u]+edg[i].val;
				if(v==now)   //关键点，返回目标点上面连接的城市号 
					return u;
				if(!vis[v])
				{
					vis[v]=1;
					q.push(v);
				}
			}
		}
	}
}
int main(){
	int M,a,b;
	scanf("%d",&M);
	while(M--){
		top=0;
		scanf("%d%d",&N,&S);
		memset(head,-1,sizeof(head));
		for(int i=0;i<N-1;i++){
			scanf("%d%d",&a,&b);
			add(a,b,1);
			add(b,a,1);
		}
		for(int i=1;i<=N;i++)
		{
			if(i!=1)
				printf(" ");
			if(i==S)
			{
				printf("-1");
				continue;
			}
			int x=SPFA(i);
			printf("%d",x);
		}
		printf("\n");
	}
	return 0;
}