专题6：并查集

我终于有增加知识的宽度了 

问题 A: 线路规划

时间限制: 1 Sec  内存限制: 128 MB
[提交] [状态]

题目描述

有n 个村庄之间需要架设通信线路，使得任意两个村庄之间均可通信。两个村庄a， b 间可通信，当且仅当它们之间存在一条通信线路或者存在村庄c 使得a，c 和b，c 间均可通信。给出村庄之间架设通信线路的代价，求出最小的总代价。

输入

第一行包含两个整数n，m，分别表示村庄数量和可以架设通信线路的村庄对数。以下m 行，每行三个整数a，b，c，表示村庄a，b之间架设线路的代价为c（村庄从0 开始编号）。

输出

一个整数，最小总代价。

样例输入 Copy

3 3
0 1 1
1 2 1
2 0 3

样例输出 Copy

2

提示

对于50% 的数据，n<=100，m <=n^2
对于全部数据，1<=n<=10^5; n-1<=m<=10^5，所有代价均在[0， 10^6] 范围内，保证问题有解。

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
typedef pair<int,int> pp;
const int mod=1e5+5;
int A[mod],f[mod];
struct road{
 int l;
 int r;
 int v;
}ro[mod];
bool cmp(struct road a,struct road b)
{
     return a.v<b.v;
}
int Find(int a)
{
    if(f[a]==a)
        return a;
    else
    {
        f[a]=Find(f[a]);
        return f[a];
    }
}
int isnfa(int a,int b)
{
    if(Find(a)!=Find(b))
    {
        f[Find(a)]=Find(b);
        return 1;
    }
    return 0;
}
int main()
{
  int n,m;
  scanf("%d%d",&n,&m);
  for(int i=1;i<=m;i++)
  {
      scanf("%d%d%d",&ro[i].l,&ro[i].r,&ro[i].v);
  }
  sort(ro+1,ro+1+m,cmp);
  for(int i=0;i<n;i++)
    f[i]=i;
  ll sum=0,cnt=0;
  for(int i=1;i<=m;i++)
  {
      if(isnfa(ro[i].l,ro[i].r))//
      {
          sum+=ro[i].v;
          cnt++;
      }
      if(cnt==n-1) break;
  }
  printf("%lld",sum);
  return 0;
}
 
/**************************************************************
    Problem: 14856
    User: 2019UPC110
    Language: C++
    Result: 正确
    Time:79 ms
    Memory:3980 kb
****************************************************************/

 

 

问题 B: 连接格点

时间限制: 1 Sec  内存限制: 128 MB
[提交] [状态]

题目描述

有一个M行N列的点阵，相邻两点可以相连。一条纵向的连线花费一个单位，一条横向的连线花费两个单位。某些点之间已经有连线了，试问至少还需要花费多少个单位才能使所有的点全部连通。

输入

第一行输入两个正整数m和n。
以下若干行每行四个正整数x1,y1,x2,y2，表示第x1行第y1列的点和第x2行第y2列的点已经有连线。输入保证|x1-x2|+|y1-y2|=1。

 

输出

输出使得连通所有点还需要的最小花费。

样例输入 Copy

2 2
1 1 2 1

样例输出 Copy

3

提示

30%数据：n*m<=1000
100%数据：m,n<=1000

#pragma GCC optimize(3)
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pp;
#define fi first
#define se second
#define pb push_back
const double eps=1e-6;
const double pi=cos(-1);
const int N=4e6+5;
const int mod=1e9+7;
int n,m,f[N];
int num(int x,int y){return (x-1)*m+y;};
struct node
{
    int u,v,w;
    bool operator <(const node&t)const
    {
        return w<t.w;
    }
}e[N];
int getf(int x)
{
    return f[x]==x?x:f[x]=getf(f[x]);
}
int main()
{
    scanf("%d%d",&n,&m);
    int x1,x2,y1,y2;for(int i=1;i<=n*m;i++)f[i]=i;
    while(~scanf("%d%d%d%d",&x1,&y1,&x2,&y2)){
        int x=getf(num(x1,y1));
        int y=getf(num(x2,y2));
        if(x==y)continue;
        f[x]=y;
    }
 
    int k=0;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            if(i<n)e[++k]={num(i,j),num(i+1,j),1};
            if(j<m)e[++k]={num(i,j),num(i,j+1),2};
        }
    }sort(e+1,e+1+k);
    int res=0;
    for(int i=1;i<=k;i++){
        int x=getf(e[i].u);
        int y=getf(e[i].v);
        if(x==y)continue;
        res+=e[i].w;
        f[x]=y;
    }cout<<res<<endl;
    return 0;
}

#include<bits/stdc++.h>
using namespace std;
const int maxn=10000010;
int m,n;
long long a,b,c,d;
long long fa[maxn];
int convert(int x,int y){
    return x*m+y-m;
}
void init(){
    for(int i=1;i<=m;i++){
        for(int j=1;j<=n;j++){
            fa[convert(i,j)]=convert(i,j);
        }
    }
}
int get(int x){
    if(fa[x]==x){
        return x;
    }
    fa[x]=get(fa[x]);
    return fa[x];
}
void merge(int x,int y){
    fa[get(x)]=get(y);
}
int main(){
    cin>>m>>n;
    init();
    long long ans=0;
    while(cin>>a>>b>>c>>d){
        merge(convert(a,b),convert(c,d));
    }
    for(int j=1;j<=n;j++){
        for(int i=1;i+1<=m;i++){
            if(get(convert(i,j))!=get(convert(i+1,j))){
                merge(get(convert(i,j)),get(convert(i+1,j)));
                ans++;
            }
        }
    }
    for(int i=1;i<=m;i++){
        for(int j=1;j+1<=n;j++){
            if(get(convert(i,j))!=get(convert(i,j+1))){
                merge(get(convert(i,j)),get(convert(i,j+1)));
                ans+=2;
            }
        }
    }
    cout<<ans;
    return 0;
}