并查集水题

题目是蓝桥杯的一道题：

问题描述

　　w星球的一个种植园，被分成 m * n 个小格子（东西方向m行，南北方向n列）。每个格子里种了一株合根植物。
　　这种植物有个特点，它的根可能会沿着南北或东西方向伸展，从而与另一个格子的植物合成为一体。


　　如果我们告诉你哪些小格子间出现了连根现象，你能说出这个园中一共有多少株合根植物吗？

输入格式

　　第一行，两个整数m，n，用空格分开，表示格子的行数、列数（1<m,n<1000）。
　　接下来一行，一个整数k，表示下面还有k行数据(0<k<100000)
　　接下来k行，第行两个整数a，b，表示编号为a的小格子和编号为b的小格子合根了。
　　格子的编号一行一行，从上到下，从左到右编号。
　　比如：5 * 4 的小格子，编号：
　　1 2 3 4
　　5 6 7 8
　　9 10 11 12
　　13 14 15 16
　　17 18 19 20

 

思路分析：

这个题一眼看过去像是连通块的题，然后想了一下好像并查集比较简单，所以就写了一个并查集。

总结这个是一个模板题

code：

#include <iostream>
#include <cstring>
using namespace std;
const int N = 1000010;

int fa[N],vi[N];
//vi这里挂羊头卖狗肉了应该是cnt
int find(int x){
    if(x==fa[x])return x;
    return fa[x]=find(fa[x]);
}
int main(){
    int n,m;
    cin>>n>>m;
    int tot=n*m;
    for(int i=1;i<=tot;i++){
        fa[i]=i;
    }
    int k;
    cin>>k;
    memset(vi,0,sizeof vi);
    for(int i=0;i<k;i++){
        int a,b;
        cin>>a>>b;
        fa[find(a)]=find(b);
    }
    for(int i=1;i<=tot;i++){
        int t=find(i);
        vi[t]++;
    }
    int res=0;
    for(int i=1;i<=tot;i++){
        if(vi[i]){
            res++;
        }
    }
    cout<<res<<endl;
}