P3201 [HNOI2009]梦幻布丁 [启发式合并][set]

P3201 [HNOI2009]梦幻布丁

题意：N个布丁摆成一行,进行M次操作.每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色.例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.

这道题是可以用数据结构合并的，然而我们可以用STL水过。用set。

对每种颜色开一个set，在set里面存下标。

合并做法就是暴力地把一个set里面的所有元素都添加进另一个set。

但是乱合并的复杂度是特别大的。我们可以使用启发式合并：把小的set合并到大的set去。复杂度可以证明是带\(\log\)的。

两个set的交换是\(O(n)\)的，所以我们用指针实现即可。

代码：

#include<bits/stdc++.h>

const int maxn = 1000005;

int n, m;
int cnt;
std::set<int> *s[maxn];
int a[100005];
int main() {
    scanf("%d %d", &n, &m);
    for(int i = 1; i <= 1000000; i++) s[i] = new std::set<int>;
    for(int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        cnt += a[i] != a[i - 1];
        s[a[i]]->insert(i);
    }
    int opt, x, y;
    while(m--) {
        scanf("%d", &opt);
        if(opt == 1) {
            scanf("%d %d", &x, &y);
            if(x == y) continue;
            if(s[x]->size() > s[y]->size()) std::swap(s[x], s[y]);
            for(std::set<int>::iterator it = s[x]->begin(); it != s[x]->end(); ++it) {
                cnt -= s[y]->count(*it - 1) + s[y]->count(*it + 1);
            }
            for(std::set<int>::iterator it = s[x]->begin(); it != s[x]->end(); ++it) {
                s[y]->insert(*it);
            }
            s[x]->clear();
        } else if(opt == 2) {
            printf("%d\n", cnt);
        }
    }
    return 0;
}