sdut 2421 PowerOutage（最小生成树）

题目：http://acm.sdut.edu.cn/sdutoj/problem.php?action=showproblem&problemid=2421

利用最小生成树求出总权值×2 - 最大的叶子节点所经路径的权值

#include <iostream>
#include<cstdio>
#include<cstring>
#define inf 0x3f3f3f
using namespace std;
int b[110][110];
void  prim(int n)
{
    int i,j,pos,min;
    int d[110];
    int vis[110];
    int max[110],h=0;
    memset(vis,0,sizeof(vis));
    memset(max,0,sizeof(max));
    for(i=1;i<=n;i++)
    {
        d[i]=b[0][i];
        if(d[i]<inf)
        {
            max[i]=d[i];
            if(h<max[i])
            h=max[i];
        }

    }
    vis[0]=1;
    long long int ans=0;
    for(i=0;i<=n;i++)
    {
        min=inf;
        for(j=1;j<=n;j++)
        {
            if(!vis[j]&&d[j]<min)
            {
                pos=j;
                min=d[j];
            }
        }
        if(min==inf)
        break;
        vis[pos]=1;
        ans+=min;
        for(j=1;j<=n;j++)
        {
            if(!vis[j]&&b[pos][j]<d[j])
            {
                d[j]=b[pos][j];
                max[j]=max[pos]+d[j];//更新j点到零的最大长度
                if(h<max[j])
                h=max[j];
            }

        }
    }
    printf("%lld\n",ans*2-h);
}
int main()
{
    int n,i,x,y,d,max;
    while(~scanf("%d",&n))
    {
        memset(b,inf,sizeof(b));
        max=0;
        for(i=0;i<n;i++)
        {
            scanf("%d%d%d",&x,&y,&d);
            if(d<b[x][y])
            {
                b[x][y]=d;
            }
            b[y][x]=b[x][y];
            if(max<x)
            max=x;
            if(max<y)
            max=y;
        }
        prim(max);
    }
    return 0;
}