杭电1879题 最小生成树

根据题意，输入四个数，最后一个数是1说明路已经修了，所以开一个调用函数进行排序：首先判断最后一个数flag是不是1，是1，则排在前面，然后根据第三个数Value升序排列；最后的工作是首先判断flag是不是1，是1，则s=s+1（是表示要修的路且s=n-1），若不是1，则sum=sum+edge[i].value（sum表示成本），s=s+1;

#include<stdio.h> #include<algorithm> using namespace std; #define max 10005 int father[max],Rank[max],a[max]; int find_set(int x) { if(x!=father[x]) father[x]=find_set(father[x]); return father[x]; } bool Union(int x,int y) { x=find_set(x); y=find_set(y); if(x==y) return 0; if(Rank[x]>Rank[y]) father[y]=x; else { if(Rank[x]==Rank[y]) Rank[y]++; father[x]=y; } return 1; } struct K { int a,b,value,flag; }; K edge[max]; bool cmp(K o1,K o2) { if(o1.flag!=o2.flag) return o1.flag>o2.flag; //根据flag降序排列 else return o1.value<o2.value; //根据value升序排列 } int main() { int i,n; while(scanf("%d",&n),n) { for(i=1; i<=max; i++) { father[i]=i; Rank[i]=0; } int s=0,sum=0; for(i=1; i<=n*(n-1)/2; i++) scanf("%d%d%d%d",&edge[i].a,&edge[i].b,&edge[i].value,&edge[i].flag); sort(edge+1,edge+1+n*(n-1)/2, cmp); for(i=1; i<=n*(n-1)/2; i++) { if(edge[i].flag==1) //已经修过，不再考虑成本 s=s+1; if(Union(edge[i].a,edge[i].b)&&edge[i].flag==0) { sum=sum+edge[i].value; s=s+1; } if(s==n-1) break; } printf("%d\n",sum); } return 0; }