暑期前专题题解---数据结构---L

        并查集的应用，可以发现，因为颜色有无数多种，我们根本不可能通过关系推导得到x和y之间的颜色关系，因此我们不能将有关系的两个节点连接在一起，我们应该将所有相同颜色的点连接在一起，那么现在的难度就在于判断两个点的颜色是相同还是我们不知道，因为这两种情况，x和y都不在同一个并查集上，因此，我们用1e5个set来记录，例如s[x] 记录的就是和x有关系的点。那么就我们发现如果x和y相同，那么和x有关系的点一定和y有关系，但是如果x和y不同，那么和x有关系的点未必和y有关系，因此在x和y不同的时候，我们只需要将fy插入到fx里面即可，其他的点因为信息不明确都记录为关系不明确的点。

         合并的时候注意可以启发式合并，集合内数小的连接到集合内数多的节点上，这样时间复杂度更加的优秀。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;
int pre[maxn];
set<int> s[maxn];
void init(int n){
    for (int i=0; i<n; i++) {
        pre[i] = i;
        s[i].insert(i);
    }
}
int find(int x){
    if (pre[x] == x) return x;
    return  pre[x] = find(pre[x]);
}
void join(int x,int y,int p){
    int fx = find(x),fy = find(y);
    if (s[fx].size() > s[fy].size()){
        if (p == 1) {
            pre[fy] = fx;
            for (auto num: s[fy]){
                s[fx].insert(num);
            }
        }
    }else{
        if (p == 1){
            pre[fx] = fy;
            for (auto num: s[fx]){
                s[fy].insert(num);
            }
        }
    }
    if (p == 2){
        s[fx].insert(fy);
        s[fy].insert(fx);
    }
}
int main(){
    char opt;
    int n,m,p,x,y;
    scanf("%d%d",&n,&m);
    init(n);
    for (int i=0; i<m; i++) {
        cin >> opt;
        scanf("%d%d",&x,&y);
        if (opt == 'A'){
            scanf("%d",&p);
            join(x, y, p);
        }else if (opt == 'Q'){
            int fx = find(x),fy = find(y);
            if (fx == fy) cout << 1 << endl;
            else{
                if (s[fx].count(fy) || s[fy].count(fx)){
                    cout << 2 << endl;
                }else{
                    cout << 3 << endl;
                }
            }
        }
    }
    return 0;
}