http://acm.hdu.edu.cn/showproblem.php?pid=1879

该题依旧是最小生成树问题,只是有一点小小的变化。。。。。做完这题,猛然发现为什么prim算法优先于Kursal了。。。。。

#include<iostream>
#include<string.h>
#include<algorithm>
#include<limits.h>
#define N 105
#define  FOR(i,s,t) for(int i=(s);i<=t;++i)
using namespace std;
int n,m;
int father[N];
struct edge{int x,y,len;
    int flag;
       }map[5005];
typedef struct edge Node;
int cmp(const void *a,const void *b)
{   return ((Node*)a)->len>((Node*)b)->len?1:-1;
 }
void make_set()
{ FOR(i,1,n)
   father[i]=i;
}
int  find_set(int a)
{    while(a!=father[a])
           a=father[a];
      return a;
 }
 void Kursal()
 {     int sum=0,ans=0;
      FOR(i,1,m)
      {
      int x=find_set(map[i].x);
      int y=find_set(map[i].y);
       if(x!=y)
       {    father[y]=x;
           if(!map[i].flag) sum+=map[i].len;
           ans++;
       }
       if(ans==n-1) break;
      }
      cout<<sum<<endl;
 }
 int main()
 {   while(cin>>n&&n)
       {     m=n*(n-1)/2;
         make_set();
            FOR(i,1,m)
            {cin>>map[i].x>>map[i].y>>map[i].len>>map[i].flag;
              if(map[i].flag)  map[i].len=0;
            }
            qsort(map+1,m,sizeof(Node),cmp);
             Kursal();
       } return 0;
     }


你可能感兴趣的:(http://acm.hdu.edu.cn/showproblem.php?pid=1879)