【ZCMU1398】工程（并查集+最短路径）

题目链接

 

解题思路：

用并查集判断两个点是否连通，如果不连通则直接输出-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=find(x);f2=find(y);
        //printf("%d %d\n",f1,f2);
        if(f1!=f2)ans=-1;
        else{
            dijkstra(x);
            ans=dis[y];
        }
        printf("%d\n",ans);
    }
    return 0;
}