畅通工程 HDU

       某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。
Output对每个测试用例，在1行里输出最小的公路总长度。
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


        
 
Huge input, scanf is recommended
书上的模板。。。，正在理解ing

#include<cstdio>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;
static const int MAX=105;
static const int INFTY=(1<<21);
static const int WHITE=0;
static const int GRAY=1;
static const int BLACK=2;

int n,M[MAX][MAX];
int prim()
{   int u,minv;
    int d[MAX],color[MAX],p[MAX];
    for(int i=1;i<=n;i++)
    {   d[i]=INFTY;
        color[i]=WHITE;
        p[i]=-1;
    }
    d[1]=0;
    while(1)
    {   minv=INFTY;
        u=-1;
        for(int i=1;i<=n;i++)
        {   if(minv>d[i]&&color[i]!=BLACK)
            {   u=i;
                minv=d[i];
            }
        }
        if(u==-1) break;
        color[u]=BLACK;
        for(int v=1;v<=n;v++)
        {   if(color[v]!=BLACK&&M[u][v]!=INFTY)
            {   if(d[v]>M[u][v])
                {   d[v]=M[u][v];
                    p[v]=u;
                    color[v]=GRAY;
                }
            } 
        } 
    }
    int sum=0;
    for(int i=1;i<=n;i++)
    {   if(p[i]!=-1) sum+=M[i][p[i]];
    }
    return sum;
}
void init()
{   int i,j;
    for(i=1;i<=n;i++)
    {   for(j=1;j<=n;j++)
        {  M[i][j]=INFTY;
        }
    }

}
int main()
{   while(scanf("%d",&n),n!=0)
    {  init();
       int x,y,v;
       for(int j=0;j<(n*(n-1))/2;j++)
       {  scanf("%d%d%d",&x,&y,&v);
          if(v<M[x][y]) M[x][y]=M[y][x]=v;
       }
       printf("%d
",prim());
    }
}