HDOJ1879(继续畅通工程)

题目链接

最小生成树的题。克鲁斯卡尔算法。

View Code 
 1 #include <stdio.h>
 2 #define N 100
 3 #define M 5000
 4 struct node
 5 {
 6     int a,b,d;
 7 }edge[M];
 8 int n;
 9 int p[N];
10 void make_set()
11 {
12     int i;
13     for(i=1;i<=n;i++)   p[i]=i;
14 }
15 int find_set(int i)
16 {
17     return i==p[i]?p[i]:(p[i]=find_set(p[i]));
18 }
19 void union_set(int i,int j)
20 {
21     i=find_set(i),j=find_set(j);
22     p[j]=i;
23 }
24 int cmp(const void *a,const void *b)
25 {
26     return ((struct node*)a)->d-((struct node*)b)->d;
27 }
28 int main()
29 {
30     int i,ans,cnt,x,y,z,f;
31     while(scanf("%d",&n)&&n)
32     {
33         make_set();
34         cnt=0;
35         for(i=0;i<n*(n-1)/2;i++)
36         {
37             scanf("%d%d%d%d",&x,&y,&z,&f);
38             edge[i].a=x,edge[i].b=y,edge[i].d=z;
39             if(f)   union_set(x,y),cnt++;
40         }
41         ans=0;
42         qsort(edge,n*(n-1)/2,sizeof(edge[0]),cmp);
43         for(i=0;cnt<n-1&&i<n*(n-1)/2;i++)
44         {
45             if(find_set(edge[i].a)==find_set(edge[i].b))    continue;
46             ans+=edge[i].d;
47             union_set(edge[i].a,edge[i].b);
48             cnt++;
49         }
50         printf("%d\n",ans);
51     }
52     return 0;
53 }

 

posted @ 2012-04-20 21:23  BeatLJ  阅读(204)  评论(0)  编辑  收藏  举报