蓝桥杯历届试题-合根植物(并查集)

蓝桥杯历届试题-合根植物是一道简单的并查集问题，并查集算法的核心是把两个点连通起来的函数和找到某个点的根节点的函数，本文把这两个函数定义成join()和find();
先写出这两个函数的原型：

int find(int x)
{
     return a[x]==x?x:a[x]=find(a[x]);//返回节点x的根节点，并且压缩路径
}

//如果看不懂压缩路径的写法可以看下面这个
int find(int x)
{
     while(a[x]!=x)
     {
         x=a[x];
     }
     return x;//返回x节点的根节点
}

void join(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
     	a[fx]=fy;//让fx的根节点变为fy
}

最后写出这道题的解法：

#include<bits/stdc++.h>
using namespace std;
int a[1000005];
int count=0;
int find(int x)
{
     return a[x]==x?x:a[x]=find(a[x]);
}
void join(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
    {
           a[fx]=fy;
           count--;
    }
}
int main()
{
    int i;
    int m,n,c;
    scanf("%d%d",&m,&n);
    c=m*n;
    count=c;
    for(i=1;i<=c;i++)
     {
         a[i]=i;
     }
    int b;
    scanf("%d",&b);
    while(b--)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        join(x,y);
    }
    printf("%d\n",count);
    return 0;
}