HDU1879 prim

还是prim

预处理那些已经修好的路，变为0,即可。

#include<stdio.h>
#include<string.h>
const int maxn = 103;
const int inf = 9999999;
struct node{
    int cost,is;
}mat[ maxn ][ maxn ];
int dis[ maxn ],vis[ maxn ];
int n,sum;

int prim(){
    memset( vis,0,sizeof(vis));
    for( int i=1;i<=n;i++ )
        dis[ i ]=mat[ 1 ][ i ].cost;
    vis[ 1 ]=1;
    for( int i=1;i<=n;i++ ){
        int t,max;
        t=-1,max=inf;
        for( int j=1;j<=n;j++ ){
            if( vis[ j ]==0 && max>dis[ j ]){
                t=j,max=dis[ j ];
            }
        }
        if( t==-1 ) return sum;
        sum+=max;
        vis[ t ]=1;
        for( int j=1;j<=n;j++ ){
            if( vis[ j ]==0 && dis[ j ]>mat[ t ][ j ].cost){
                dis[ j ]=mat[ t ][ j ].cost;
            }
        }
    }
    return sum;
}

int main(){
    while( scanf("%d",&n)!=EOF,n ){
        int nn=n*(n-1)/2;
        int a,b,c,d;
        sum=0;
        memset( vis,0,sizeof( vis ));
        while( nn-- ){
            scanf("%d%d%d%d",&a,&b,&c,&d);
            mat[ a ][ b ].cost=mat[ b ][ a ].cost=c;
            mat[ a ][ b ].is=mat[ b ][ a ].is=d;
            if( d==1 ){
                //sum+=c;
                mat[ a ][ b ].cost=mat[ b ][ a ].cost=0;
            }
        }
        int ans=prim();
        printf("%d\n",ans);
    }
    return 0;
}