最小生成树

hdu 2682 最小生成树

分类： 练习 2013-05-09 08:00 152人阅读 评论(0)收藏 举报

ACMC算法图论最小生成树

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

题目大意：T组测试实例，每组实例有n个城市，每个城市有一个幸福指数V；如果 Va 或 Vb 或 Va+Vb 是素数的话，两个城市可以连接起来；将两个城市连接起来所需的花费为min(min(Va,Vb),|Va-Vb|)；如果所有的城市都能连接起来的话输出最小花费，否则输出“-1”；

[cpp] view plaincopyprint?

1. #include<iostream>
2. #include<stdio.h>
3. #include<string.h>
4. #include<stdlib.h>
5. #include<math.h>
6. #include<algorithm>
7. #define MAX 1000006
8. #define min(a,b) ((a)<(b)?(a):(b))
9. using namespace std;
10. int prime[MAX];
11. int set[606];
12. struct node
13. {
14. int a,b,len;
15. }map[180000];
17. int cmp(constvoid* a,constvoid*b)
18. {
19. return (*(node*)a).len-(*(node*)b).len;
20. }
22. //并查集
23. int find(int x)
24. {
25. if(set[x]!=x){
26. set[x]=find(set[x]);
27. }
28. return set[x];
29. }
30. int Union(int a,int b)
31. {
32. int x=find(a);
33. int y=find(b);
34. if(x==y)
35. return 0;
36. set[x]=y;
37. return 1;
38. }
40. int main()
41. {
42. int T,n,i,j,k,temp,cnt,cost;
44. //筛选法求素数
45. memset(prime,0,sizeof(prime));
46. prime[0]=1;
47. prime[1]=1;
48. for(i=2;i<MAX;i++){
49. if(prime[i]==0){
50. for(j=2;j<MAX;j++){
51. temp=i*j;
52. if(temp>MAX)break;
53. prime[temp]=1;
54. }
55. }
56. }
57. scanf("%d",&T);
58. while(T--){
59. scanf("%d",&n);
60. for(i=0;i<n;i++){
61. scanf("%d",&set[i]);
62. }
64. //建图
65. for(i=0,k=0;i<n;i++){
66. for(j=i+1;j<n;j++){
67. if(prime[set[i]]==0 || prime[set[j]]==0 || prime[set[i]+set[j]]==0){
68. map[k].a=i;
69. map[k].b=j;
70. map[k++].len=min(min(set[i],set[j]),abs(set[i]-set[j]));
71. }
72. }
73. }
74. qsort(map,k,sizeof(map[0]),cmp);
76. for(i=0;i<n;i++){
77. set[i]=i;
78. }
79. for(i=0,cnt=0,cost=0;i<k;i++){
80. if(Union(map[i].a,map[i].b)==1){
81. cost+=map[i].len;
82. cnt++;
83. }
84. }
85. if(cnt==(n-1))
86. printf("%d\n",cost);
87. else
88. printf("-1\n");
89. }
90. return 0;
91. }