有N个城市,这N个城市间只有N-1条路把这个N个城市连接起来。现在,小明在第S号城市,他有张该国地图,他想知道如果自己要去参观第T号城市,必须经过的前一个城市是几号城市(假设你不走重复的路)。
有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)
我是想用邻接表做,我用了两种方法,一种是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我还不会用,下次一定补上!
版权声明:本文为博主原创文章,未经博主允许不得转载。
原文地址:http://blog.csdn.net/nvliba/article/details/47984329