题面:
Problem G: 漂洋过海来看你
Description
BMan和hyx住在一个神奇的国度,这个国度有N个城市,这N个城市间只有N-1条路把这个N个城市连接起来。 现在BMan在第S号城市,他经常想起那个一起AC过的队友hyx,记忆它总是慢慢的累积,在他心中无法抹去,可是他并不知道hyx 在哪个城市,所以他决定用尽半年的积蓄漂洋过海去找hyx,现在BMan很想知道如果他想去hyx所在的第X号城市,必须经过的前 一个城市是第几号城市(放心,由于心系队友,BMan是不会选择走重复的路的~)
Input
第一行输入一个整数T表示测试数据共有T(1<=T<=10)组 每组测试数据的第一行输入一个正整数N(1<=N<=1000)和一个正整数S(1<=S<=1000),N表示城市的总数,S是BMan所在城市的编号 随后的N-1行,每行有两个正整数a,b(1<=a,b<=N),表示第a号城市和第b号城市之间有一条路连通。
Output
每组测试数据输N个正整数,其中,第i个数表示从S走到i号城市,必须要经过的上一个城市的编号(其中i=S时,请输出-1)
Sample Input
110 11 91 88 1010 38 61 210 49 53 7
Sample Output
-1 1 10 10 9 8 3 1 1 8
题意:
给定一张图,求源点到每个点最短路的路径上,该点的前一个点的序号,若没有则为-1
#include<iostream>
#define inf 100000
using namespace std;
//map存图,n为规模,pre保存该点前一个点,dist为 源点到每个点的最短距离
int map[1005][1005],n,pre[1005],dist[1005];
//mark访问标记
bool mark[1005];
void dijk(int u)//u为出发点
{
int i,j,mins,v;
for(int i=1;i<=n;i++)
{
dist[i]=map[u][i];
mark[i]=false;
pre[i]=u;//初始化为源点
}
mark[u]=true;
dist[u]=0;
while(1)
{
mins=inf;
for(int j=1;j<=n;j++)
{
if(!mark[j]&&dist[j]<mins)//遍历寻找当前的最短路
mins=dist[j],v=j;
}
if(mins==inf)break;//已经没有路了
mark[v]=true;
for(int j=1;j<=n;j++)
{
if(!mark[j]&&dist[v]+map[v][j]<dist[j])//比当前路短
{
dist[j]=dist[v]+map[v][j];
pre[j]=v;//记录该点的前一个点
}
}
}
}
int main()
{
int t,s,x,y;
cin>>t;
while(t--)
{
cin>>n>>s;
for(int i=1;i<=n;i++)//初始化,也可分离为函数
{
for(int j=1;j<=n;j++)
map[i][j]=inf;
}
for(int i=1;i<n;i++)
{
cin>>x>>y;
map[x][y]=map[y][x]=1;//假定通路长度为1
}
dijk(s);
pre[s]=-1;//源点的前缀为-1
cout<<pre[1];
for(int i=2;i<=n;i++)
cout<<" "<<pre[i];
cout<<endl;
}
return 0;
}