【PTA】公路村村通 (30 分)

 

#include <iostream>
#define MAXN 1010
#define MAXDATA 1000000000
using namespace std;

int N,M;
int G[MAXN][MAXN];

int dis[MAXN];
int vis[MAXN];
int prim(){
    int ans=0;//初始化最小边权之和ans
    fill(dis,dis+N,MAXDATA);
    dis[0]=0;//定义起始点为0
    int i;
    for(i=0;i<N;i++){
        //1:FindMin
        int minId=-1,minValue=MAXDATA;
        for(int j=0;j<N;j++){
            if(vis[j]==0 && dis[j]<minValue){
                minValue=dis[j];
                minId=j;
            }
        }
        if(minId==-1) break;//只有没有最小生成树,才可能中途退出
        vis[minId]=1;
        ans+=dis[minId];
        //2:邻接点
        for(int u=0;u<N;u++){
            if(vis[u]==0 && G[minId][u]!=MAXDATA){//未收录的邻接点
                if(dis[u]>G[minId][u]){
                    dis[u]=G[minId][u];
                }
            }
        }
    }
    if(i<N) return -1;
    else    return ans;
}

int main(){
    scanf("%d %d",&N,&M);
    for(int i=0;i<N;i++){
        for(int j=0;j<N;j++){
            G[i][j]=MAXDATA;
        }
    }

    int tmp1,tmp2,tmpl;
    for(int i=0;i<M;i++){
        scanf("%d %d %d",&tmp1,&tmp2,&tmpl);
        G[tmp1-1][tmp2-1]=tmpl;
        G[tmp2-1][tmp1-1]=tmpl;
    }

    printf("%d",prim());


    return 0;
}