问题 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;
}
而spfs版本是把和当前相连的搜索完,在进行下一个!
都很巧妙,然而vector我还不会用,下次一定补上!