蓝桥杯T5合根植物——并查集模板题

5.合根植物 - 蓝桥云课 (lanqiao.cn) 

#include <bits/stdc++.h>
using namespace std;
int m, n, pre[1000000];
set<int>s;
int find(int x) {
    if (pre[x] == x) return x;
    return find(pre[x]);
}
int main()
{
    // 请在此输入您的代码
    cin >> m >> n;
    for (int i = 1; i <= m * n; i++)
        pre[i] = i;
    int k, a, b;
    cin >> k;
    for (int i = 1; i <= k; i++) {
        cin >> a >> b;
        int x=find(a),y=find(b);
        if(x!=y) pre[x] = y;
    }
    for (int i = 1; i <= n * m; i++) {
        s.insert(find(i));
    }    
    cout << s.size();
    return 0;
}