hdu 1879 简单最小生成树

如果已经建好的公路就直接连一权值为0的边

#include<stdio.h>
#include<string.h>
#define inf 999999
int flag[200];
int D[200];
int map[200][200];
int n,i,j;
int prime()//下标1开始 
{
    int v,k;
    int ret=0,mi;
    for(i=1;i<=n;i++){
        flag[i]=0;
        D[i]=inf;
    }D[1]=0;flag[1]=1;v=1;
    for(k=1;k<n;k++){
        for(i=1;i<=n;i++)if(!flag[i]){
            if(map[v][i]<D[i])
                D[i]=map[v][i];
        }
        mi=inf;
        for(i=1;i<=n;i++)
            if(!flag[i]&&D[i]<mi)
                mi=D[v=i];
            flag[v]=1;
            ret+=mi;
    }
    return ret;
}
int main()
{
    while(scanf("%d",&n),n)
    {
        int a,b,c,w;
        for(i=0;i<n*(n-1)/2;i++)
        {
            scanf("%d%d%d%d",&a,&b,&w,&c);
            if(c==0) map[a][b]=map[b][a]=w;
            else map[a][b]=map[b][a]=0;
        }
        printf("%d\n",prime());
    }
    return 0;
}