hdu 1233 还是畅通工程(最小生成树)

思路:最小生成树模板题。

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;

const int INF=0x3f3f3f3f;
const int MAXN=110;
bool vis[MAXN];
int lowc[MAXN];

int Prim(int cost[][MAXN],int n)
{
    int ans=0;
    memset(vis,false,sizeof(vis));
    vis[0]=true;
    for(int i=1; i<n; i++)lowc[i]=cost[0][i];
    for(int i=1; i<n; i++)
    {
        int minc=INF;
        int p=-1;
        for(int j=0; j<n; j++)
            if(!vis[j]&&minc>lowc[j])
            {
                minc=lowc[j];
                p=j;
            }
        if(minc==INF)return -1;//原图不连通
        ans+=minc;
        vis[p]=true;
        for(int j=0; j<n; j++)
            if(!vis[j]&&lowc[j]>cost[p][j])
                lowc[j]=cost[p][j];
    }
    return ans;
}

int main()
{
    int cost[MAXN][MAXN];
    int N;
    int u,v,w;
    while(~scanf("%d",&N))
    {
        if(N==0)break;
        for(int i=0; i<MAXN; ++i)
        {
            for(int j=0; j<MAXN; ++j)
            {
                cost[i][j]=INF;
            }
        }

        int M=N*(N-1)/2;
        for(int i=0; i<M; ++i)
        {
            scanf("%d%d%d",&u,&v,&w);
            --u;
            --v;
            cost[u][v]=cost[v][u]=w;
        }

        printf("%d\n",Prim(cost,N));
    }
    return 0;
}

你可能感兴趣的:(hdu 1233 还是畅通工程(最小生成树))