【进阶课】启发式合并

启发式合并

---

问题概述： 有 $n$ 个元素，分成 $m$ 个集合。依次操作对 $m$ 个集合进行两两合并。

算法思想： 每次合并都把小集合中的元素往大集合里放。对每个元素来说，所在集合个数是指数增长的，这样合并的移动次数是 $O(logm)$ 级别的。对于所有元素，移动次数是 $O(nlogm)$ 级别的。

博客链接：启发式合并

---

Acwing 2154 - 梦幻布丁

• i是颜色，p[i]是颜色在序列中的编号。必要时可以交换两个颜色的编号。注意如何更新颜色段数量。

AC代码：

#include<bits/stdc++.h>
using namespace std;

typedef long long LL;
const int N = 1e5 + 10, M = 1e6 + 10;

int n, m;
int col[N], p[M];
vector<int>idx[M];
int ans;

void merge(int & x, int & y)
{
    if(x == y) return ;
    if(idx[x].size() > idx[y].size()) swap(x, y);
    for(int i : idx[x]){
        ans -= (col[i - 1] == y) + (col[i + 1] == y);
    }
    for(int i : idx[x]){
        col[i] = y;
        idx[y].push_back(i);
    }
    idx[x].clear();
}

int main()
{
    cin >> n >> m;
    for(int i=1; i<=n; i++){
        cin >> col[i];
        idx[col[i]].push_back(i);
        if(col[i] != col[i - 1]) ans ++ ;
    }

    for(int i=0; i<M; i++) p[i] = i;

    while(m -- ){
        int op; cin >> op;
        if(op == 2) cout << ans << '\n';
        else{
            int x, y; cin >> x >> y;
            merge(p[x], p[y]);
        }
    }

    system("pause");
    return 0;
}