zoukankan      html  css  js  c++  java

  • HDOJ3371 Connect the Cities[kruskal算法]

    Connect the Cities

    Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
    Total Submission(s): 4439    Accepted Submission(s): 1298


    Problem Description
    In 2100, since the sea level rise, most of the cities disappear. Though some survived cities are still connected with others, but most of them become disconnected. The government wants to build some roads to connect all of these cities again, but they don’t want to take too much money.  
     
    Input
    The first line contains the number of test cases.
    Each test case starts with three integers: n, m and k. n (3 <= n <=500) stands for the number of survived cities, m (0 <= m <= 25000) stands for the number of roads you can choose to connect the cities and k (0 <= k <= 100) stands for the number of still connected cities.
    To make it easy, the cities are signed from 1 to n.
    Then follow m lines, each contains three integers p, q and c (0 <= c <= 1000), means it takes c to connect p and q.
    Then follow k lines, each line starts with an integer t (2 <= t <= n) stands for the number of this connected cities. Then t integers follow stands for the id of these cities.
     
    Output
    For each case, output the least money you need to take, if it’s impossible, just output -1.
     
    Sample Input
    1 6 4 3 1 4 2 2 6 1 2 3 5 3 4 33 2 1 2 2 1 3 3 4 5 6
     
    Sample Output
    1
     
    Author
    dandelion
     
    Source
    HDOJ Monthly Contest – 2010.04.04
     
    Recommend
    lcy
     
     
     
     
     
     
     

    并查集的记忆化搜索
    code:
     1 #include <iostream>   
 2 #include <iomanip>   
 3 #include <fstream>   
 4 #include <sstream>   
 5 #include <algorithm>   
 6 #include <string>   
 7 #include <set>   
 8 #include <utility>   
 9 #include <queue>   
10 #include <stack>   
11 #include <list>   
12 #include <vector>   
13 #include <cstdio>   
14 #include <cstdlib>   
15 #include <cstring>   
16 #include <cmath>   
17 #include <ctime>   
18 #include <ctype.h> 
19 using namespace std;
20 const int N=1000000;
21 int Tc,n,m,k,done,ans;
22 int father[N];
23 int num[N];
24 
25 int findset(int p){
26     if (father[p]==p)
27         return p;
28     father[p]=findset(father[p]);
29     return father[p];
30 }
31 
32 typedef struct edge
33 {
34     int x,y;
35     int len;
36 }Edge;
37 Edge edge[N];
38 
39 int cmp(const Edge &a,const Edge &b){
40     return a.len<b.len;
41 }
42 
43 void kruskal(){
44     if (done==n-1) return;
45     sort(edge+1,edge+m+1,cmp);
46     for (int i=1;i<=m;i++){
47         int x=edge[i].x,y=edge[i].y;
48         if (findset(x)!=findset(y)){
49             if (num[father[x]]<num[father[y]]){
50                 num[father[y]]+=num[father[x]];
51                 father[father[x]]=father[y];
52             }
53             else{
54                 num[father[x]]+=num[father[y]];
55                 father[father[y]]=father[x];
56             }
57             done++;
58             ans+=edge[i].len;
59             if (done==n-1) return;
60         }
61     }
62 }
63 
64 int main(){
65     scanf("%d",&Tc);
66     for (int q=1;q<=Tc;q++)
67     {
68         done=0; ans=0;
69         int i,j,t,one,x;
70         scanf("%d%d%d",&n,&m,&k);
71         for (i=1;i<=n;i++){
72             father[i]=i;
73             num[i]=1;
74         }
75         for (i=1;i<=m;i++)
76             scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].len);
77         for (i=1;i<=k;i++){
78             scanf("%d%d",&t,&one);
79             for (j=1;j<t;j++){
80                 scanf("%d",&x);
81                 if (findset(x)!=findset(one)){
82                     if (num[father[x]]<num[father[one]]){
83                         num[father[one]]+=num[father[x]];
84                         father[father[x]]=father[one];
85                     }
86                     else{
87                         num[father[x]]+=num[father[one]];
88                         father[father[one]]=father[x];
89                     }
90                     done++;
91                 }
92             }
93         }
94         kruskal();
95         if (done==n-1) printf("%d\n",ans);
96         else printf("-1\n");
97     }
98 }

    还可以水过去

    code:

     1 #include<iostream>
 2 #include<algorithm>
 3 using namespace std;
 4 int f[505],L[505];
 5 struct node
 6 {
 7     int x,y,s;
 8 }dd[25005];
 9 bool cmp(node a,node b)
10 {
11     return a.s<b.s;
12 }
13 int find(int a)
14 {
15     if(a!=f[a])
16         a=find(f[f[a]]);
17     return a;
18 }
19 void Union(int x,int y)
20 {
21     if(x>y)
22         f[x]=y;
23     else
24         f[y]=x;
25 }
26 int main()
27 {
28     int n,m,k,t;
29     int i,j;
30     int T;
31     scanf("%d",&T);
32     while(T--)
33     {
34         scanf("%d%d%d",&n,&m,&k);
35         for(i=0;i<m;i++)
36             scanf("%d %d %d",&dd[i].x,&dd[i].y,&dd[i].s);
37         sort(dd,dd+m,cmp);
38         for(i=1;i<=n;i++)
39             f[i]=i;
40         int One,Two;
41         for(i=0;i<k;i++)
42         {
43             scanf("%d%d",&t,&One);
44             for(j=1;j<t;j++)
45             {
46                 scanf("%d",&Two);
47                 int xx=find(One);
48                 int yy=find(Two);
49                 if(xx!=yy)
50                     Union(xx,yy);
51             }
52         }
53         int sum=0;
54         for(i=0;i<m;i++)
55         {
56             int xx=find(dd[i].x);
57             int yy=find(dd[i].y);
58             if(xx!=yy)
59             {
60                 Union(xx,yy);
61                 sum+=dd[i].s;
62             }
63         }
64         int cnt=0;
65         for(i=1;i<=n;i++)
66             if(i==f[i])
67                 cnt++;
68         if(cnt>1)
69             printf("-1\n");
70         else
71             printf("%d\n",sum);
72     }
73     return 0;
74 }

    以树的数量进行记录

    code:

      1 #include<cstdio>
  2 #include<cstring>
  3 #include<cstdlib>
  4 struct node
  5 {
  6     int x,y,z;
  7 }arr[25100];
  8 int n,m,k,t,a[510],pre[510],cnt;
  9 bool v[510];
 10 int cmp(const void *p,const void *q)
 11 {
 12     return ((node *)p)->z-((node *)q)->z;
 13 }
 14 int find(int x)
 15 {
 16     while(x!=pre[x])
 17         x=pre[x];
 18     return x;
 19 }
 20 void bcj()
 21 {
 22     int ans=0,flag=0,cat=0;
 23     bool vit[510],v[510];
 24     for(int i=1;i<=n;i++)
 25         vit[i]=false;
 26     for(int i=0;i<m;i++)
 27     {
 28         int fx=find(arr[i].x);
 29         int fy=find(arr[i].y);
 30         if(fx!=fy)
 31         {
 32             pre[fx]=fy;
 33             cat++;
 34             ans+=arr[i].z;
 35         }
 36         if(cat==cnt-1)
 37             break;
 38     }
 39     for(int j=1;j<=n;j++)
 40         vit[find(j)]=true;
 41     for(int j=1;j<=n;j++)
 42     {
 43         if(vit[j])
 44         {
 45             flag++;
 46             vit[j]=false;
 47         }
 48     }
 49     if(flag==1)
 50     {
 51         printf("%d\n",ans);
 52         return;
 53     }
 54     else
 55     {
 56         printf("-1\n");
 57         return;
 58     }
 59 }
 60 int main()
 61 {
 62     int p;
 63     while(~scanf("%d",&p))
 64     {
 65         while(p--)
 66         {
 67             cnt=0;
 68             scanf("%d%d%d",&n,&m,&k);
 69             for(int i=1;i<=n;i++)
 70             {
 71                 pre[i]=i;v[i]=false;
 72             }
 73             for(int i=0;i<m;i++)
 74                 scanf("%d%d%d",&arr[i].x,&arr[i].y,&arr[i].z);
 75             qsort(arr,m,sizeof(arr[0]),cmp);
 76             for(int i=0;i<k;i++)
 77             {
 78                 scanf("%d",&t);
 79                 for(int j=1;j<=t;j++)
 80                     scanf("%d",&a[j]);
 81                 for(int j=1;j<t;j++)
 82                 {
 83                     int fx=find(a[j]);
 84                     int fy=find(a[j+1]);
 85                     if(fx!=fy)
 86                         pre[fx]=fy;
 87                 }
 88             }
 89             for(int i=1;i<=n;i++)
 90             {
 91                 int fx=find(i);
 92                 if(!v[fx])
 93                 {
 94                     v[fx]=true;
 95                     cnt++;
 96                 }
 97             }
 98             //printf("cnt=%d\n",cnt);
 99             bcj();
100         }
101     }
102     return 0;
103 }

    注意：

    刚开始的时候，findset()这个函数一直是不对，以后要改过来！

    正确写法：

    1 int findset(int p){
2     if (father[p]==p)
3         return p;
4     father[p]=findset(father[p]);
5     return father[p];
6 }
  • 相关阅读:
    C#的泛型的类型参数可以有带参数的构造函数的约束方式吗
    Object Pool
    链表排序（冒泡、选择、插入、快排、归并、希尔、堆排序）
    ASP.NET MVC应用程序使用axd格式文件
    框架设计--服务总线
    Mylyn
    不要问我能赚多少，当你问这句话的时候，你的思想还停留在给别人打工的阶段，我只是你的仓库，能挣多少得问你自己想挣多少(转)
    maven中如果使用本地jar
    君子和而不同，小人同而不和
    失去控制，并随之失去指挥
  • 原文地址：https://www.cnblogs.com/XBWer/p/2623877.html
Copyright © 2011-2022 走看看