2016HUAS_ACM暑假集训3G

最小生成树，题目简单。套的Prim模板，其他的题目比较有意义。

Sample Input

3                             //村庄个数
1 2 1                       //连通情况及权值
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0

Sample Output       //输出最小生成树的权

3

5

#include<stdio.h>
#include<string.h>
#define size 100
#define inf 0x3f3f3f3f

int vellage[size][size],flag[size];
int N,T,a,b,c,i,j,k,ans,maxn;

void Init()
{
    T=N*(N-1)>>1;
    memset(flag,0,sizeof(flag));
    memset(vellage,0,sizeof(vellage));
}    

int Prim()
{
    ans=0;
    for(int h=1;h<=N;h++)
    {
        maxn=inf;
        for(i=1;i<=N;i++)
        {
            if(flag[i]!=1&&vellage[1][i]<maxn)
            {
                maxn=vellage[1][i];
                k=i;
            }
        }
        ans+=maxn;
        flag[k]=1;
        for(j=2;j<=N;j++)
        {
            if(flag[j]!=1&&vellage[k][j]<vellage[1][j])
                vellage[1][j]=vellage[k][j];
        }
    }
    return ans;
}

int main()
{
    while(scanf("%d",&N),N)
    {
        Init();
        while(T--)
        {
            scanf("%d %d %d",&a,&b,&c);
            vellage[a][b]=vellage[b][a]=c;

        }
        printf("%d
",Prim());
    }
    return 0;
}