HDU 1879 继续畅通工程

题目链接http://acm.hdu.edu.cn/showproblem.php?pid=1879

还是最小生成树的模板题

我用的Prim算法

#include<stdio.h>
 #include<string.h>
 int map[110][110],lowcost[110],visit[110];
 #define N 1000000
 int prim(int maxminum)
 {
     int i,sum,j,k,min;
     for(i=1;i<=maxminum;i++)
     {
         lowcost[i]=map[1][i];
         visit[i]=0;
     }
     sum=0;
     visit[1]=1;
     for(i=1;i<maxminum;i++)
     {
         min=N;
         for(j=1;j<=maxminum;j++)
         {
             if(visit[j]==0&&lowcost[j]<min)
             {
                 min=lowcost[j];
                 k=j;
             }
         }
        
         if(min==N)
         break;
   visit[k]=1;
         sum+=min;
         for(j=1;j<=maxminum;j++)
         {
             if(visit[j]==0&&map[k][j]<lowcost[j])
             {
                 lowcost[j]=map[k][j];
             }
         }
     }
     return sum;
 }
 int main()
 {
     int maxmax,maxminum,i,a,b,value,sum,flag;
     while(~scanf("%d",&maxminum))
     {
         if(maxminum==0)
         break;
         memset(map,N,sizeof(map));
         maxmax=(maxminum*(maxminum-1))/2;
   for(i=1;i<=maxminum;i++)
             map[i][i]=0;
         for(i=1;i<=maxmax;i++)
         {
             scanf("%d%d%d%d",&a,&b,&value,&flag);
    if(flag==0)
             {
     if(value<map[a][b])
             {
                 map[a][b]=map[b][a]=value;
             }
    }
    else
    {
     map[a][b]=map[b][a]=0;
    }
         }
         sum=prim(maxminum);
         printf("%d\n",sum);
     }
     return 0;
 }