解题思路:
用并查集判断两个点是否连通,如果不连通则直接输出-1,若连通则通过dijkstra最短路径算法计算两点直接的最短路径。
代码:
#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
int pre[1005],cost[1005][1005],vis[1005],dis[1005];
int n,m;
int find(int x)
{
int r=x;
while(r!=pre[r])r=pre[r];
int j=x,i;
while(j!=r){
i=pre[j];
pre[j]=r;
j=i;
}
return r;
}
void dijkstra(int a)
{
int i,j;
for(i=0;i<=n-1;i++){
vis[i]=0;
dis[i]=INF;
}
dis[a]=0;
while(1){
int v=-1;
for(i=0;i<=n-1;i++){
if(!vis[i] && (v==-1 ||dis[i]<dis[v])){
v=i;
}
}
if(v==-1)break;
vis[v]=1;
for(i=0;i<=n-1;i++)dis[i]=min(dis[i],dis[v]+cost[v][i]);
}
}
int main()
{
while(~scanf("%d%d",&n,&m)){
int u,v,w,f1,f2,ans=0,i,x,y;
memset(cost,INF,sizeof(cost));
for(i=0;i<=n-1;i++)pre[i]=i;
while(m--){
scanf("%d%d%d",&u,&v,&w);
cost[u][v]=w;cost[v][u]=w;
f1=find(u);
f2=find(v);
if(f1!=f2)pre[f2]=f1;
}
scanf("%d%d",&x,&y);
f1=pre[x];f2=pre[y];
//printf("%d %d\n",f1,f2);
if(f1!=f2)ans=-1;
else{
dijkstra(x);
ans=dis[y];
}
printf("%d\n",ans);
}
return 0;
}