【HPU】[1736]老王修马路(二)

最小生成树问题
没什么弯
需要注意的是有可能无法连通

#include<stdio.h>
#include<algorithm>
using namespace std;
int main() {
    int T;
    scanf("%d",&T);
    while(T--) {
        int dis[120];
        int e[120][120];
        bool flag[120];
        int inf=999999;
        int n,m;
        scanf("%d %d",&n,&m);
        for(int i=1; i<=n; i++) {
            dis[i]=inf;
            flag[i]=false;
        }
        for(int i=1; i<=n; i++) {
            for(int j=1; j<=n; j++) {
                e[i][j]=inf;
            }
        }
        while(m--) {
            int i,j,t;
            scanf("%d %d %d",&i,&j,&t);
            if(e[i][j]>t) {
                e[i][j]=t;
                e[j][i]=t;
            }
        }
        int sum=0;
        dis[1]=0;
        bool x=true;
        while(true) {
            int v=-1;
            for(int u=1; u<=n; u++) {
                if(!flag[u]&&(v==-1||dis[u]<dis[v]))
                    v=u;
            }
            if(v==-1)
                break;
            else {
                flag[v]=true;
                if(dis[v]==inf) {
                    x=false;
                    printf("Wrong\n");
                    break;
                } else
                    sum+=dis[v];
                for(int u=1; u<=n; u++) {
                    dis[u]=min(dis[u],e[v][u]);
                }
            }
        }
        if(x)
            printf("%d\n",sum);
    }
    return 0;
}

题目地址:【HPU】[1736]老王修马路(二)