HDU 1879 继续畅通工程 最小生成树P算法

3
1 2 1 0
1 3 2 0
2 3 4 0
3
1 2 1 0
1 3 2 0
2 3 4 1
3
1 2 1 0
1 3 2 1
2 3 4 1
0

3
1
0

/*
HDOJ 1879 最小生成树P算法 
*/ 

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

#define INF 100000000
const int M=110;

int n,map[M][M];
int vis[M],dis[M];

void Prime()
{
    int i,j,sum,min,loc;
    
    memset(vis,0,sizeof(vis));
    for(j=1;j<=n;j++)
    {
        if(j==1)
            dis[j]=0;
        else
            dis[j]=map[1][j];
    }    
    
    sum=0;
    for(i=1;i<=n;i++)
    {
        min=INF;
        for(j=1;j<=n;j++)
        {
            if(!vis[j]&&dis[j]<min)
            {
                min=dis[j];
                loc=j;
            }    
        }
        vis[loc]=1;
        sum+=min;
        for(j=1;j<=n;j++)
        {
            if(!vis[j]&&map[loc][j]<dis[j])
                dis[j]=map[loc][j];
        }
    }
    printf("%d\n",sum);
} 

int main()
{
    int i,j,a,b,v,f,m;
    
    //freopen("test.txt","r",stdin);
    while(scanf("%d",&n),n)
    {            
        m=n*(n-1)/2;        
        for(i=0;i<m;i++)
        {
            scanf("%d%d%d%d",&a,&b,&v,&f);
            if(f==1)//已通无需修建 
                map[a][b]=map[b][a]=0;
            else
            {
                map[a][b]=map[b][a]=v;
            }
        }
        Prime();
    }
    return 0;
}