pta 08-图7 公路村村通 (30分) 最小生成树

#include<bits/stdc++.h>
using namespace std;
const int inf=0x7fffffff;

int n,m,g[1024][1024],dist[1024];

void prim()
{
    int ans,i,j,v,minx;
    for(i=0; i<n; i++)
        dist[i]=g[0][i];
    dist[0]=-1;
    ans=0;
    for(i=1; i<n; i++)
    {
        minx=inf;
        v=-1;
        for(j=0; j<n; j++)
        {
            if(dist[j]<minx&&dist[j]!=-1)
            {
                minx=dist[j];
                v=j;
            }
        }
        if(v!=-1)
        {
            ans+=dist[v];
            dist[v]=-1;
            for(j=0; j<n; j++)
                if(g[v][j]<dist[j])
                    dist[j]=g[v][j];
        }
    }
    for(i=0;i<n;i++)
        if(dist[i]!=-1) break;

    if(i<n) ans=-1;
    printf("%d\n",ans);
}

int main()
{
    int i,j,u,v,w;
    while(~scanf("%d%d",&n,&m))
    {
        for(i=0; i<n; i++)
            for(j=0; j<n; j++)
                if(i==j) g[i][j]=0;
                else g[i][j]=inf;

        for(i=0; i<m; i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            u--;
            v--;
            g[u][v]=g[v][u]=min(g[u][v],w);
        }
        prim();
    }
    return 0;
}