[HDU]3371 Connect the Cities
View Code
1 #include<stdio.h>
2 #include<stdlib.h>
3 struct data{
4 int st,fi,v;
5 }a[25010];
6 int i,j,res,ans,file,t,w1,w,father1,father2,father[510],n,m,k,flag;
7 void qsort1(int l,int r)
8 {
9 int i=l,j=r,t=a[(l+r)/2].v;
10 struct data change;
11 do{
12 while (a[i].v<t) i++;
13 while (a[j].v>t) j--;
14 if (i<=j)
15 {
16 change=a[i];
17 a[i]=a[j];
18 a[j]=change;
19 i++;
20 j--;
21 }
22 }while (i<=j);
23 if (l<j)qsort1(l,j);
24 if (i<r)qsort1(i,r);
25 }
26 int getfather(int i)
27 {
28 if (father[i]!=i)return father[i]=getfather(father[i]);
29 return i;
30 }
31 int main()
32 {
33 scanf("%d",&file);
34 while (file--)
35 {
36 flag=res=ans=0;
37 scanf("%d %d %d",&n,&m,&k);
38 for (i=1;i<=n;i++)father[i]=i;
39 for (i=1;i<=m;i++)
40 scanf("%d %d %d",&a[i].st,&a[i].fi,&a[i].v);
41 for (i=1;i<=k;i++)
42 {
43 scanf("%d",&t);
44 if (t>0)
45 {
46 scanf("%d",&w);
47 father1=getfather(w);
48 for (j=2;j<=t;j++)
49 {
50
51 scanf("%d",&w1);
52 father[getfather(w1)]=father1;
53 }
54 }
55 }
56 for (i=1;i<=n;i++)
57 if (father[i]!=i)
58 {
59 res++;
60 }
61 qsort1(1,m);
62 i=1;
63 while (res<n-1)
64 {
65 int father1=getfather(a[i].st);
66 int father2=getfather(a[i].fi);
67 if (father1!=father2)
68 {
69 father[father1]=father2;
70 ans+=a[i].v;
71 res++;
72 }
73 i++;
74 if (i>m)
75 {
76 printf("-1\n");
77 flag=1;
78 break;
79 }
80 }
81 if (flag)continue;
82 printf("%d\n",ans);
83 }
84 return 0;
85 }
题目大意：n个城市，中间可建立m条路，且其中有k组城市已联通，问将所有城市都联通的最小代价。
大体思路：相当于是求一个部分路权为0的最小生成树。可以用Kruskal算法+并查集+排序优化。如果用并查集做的话应该可以直接将n组城市分别合到n个集合中。
注意事项：并查集合并时取“最父亲”的那个父亲，即root。
