BZOJ 1483|HNOI 2009|梦幻布丁|Set

给定初始序列，每个元素有个颜色，查询不同颜色块的个数，或合并两个颜色。

合并操作要把小的合并到大的上去，这样连续区间的长度至少会翻倍，复杂度$O(n\log n)$
但是我用的是set。。应该是$O(n\log^2n)$了。。

#include <cstdio>
#include <set>
#include <algorithm>
using namespace std;
const int N = 1000005;
#define FOR(i,j,k) for(i=j;i<=k;++i)
#define foreach(i,j) for(typeof(j.begin()) i=j.begin();i!=j.end();++i)
int fa[N], c[N], ans = 0;
set<int> h[N];
int main() {
    int i, n, m, f, a, b;
    scanf("%d%d", &n, &m);
    FOR(i,1,n) scanf("%d", &c[i]);
    FOR(i,1,n) {
        fa[c[i]] = c[i];
        h[c[i]].insert(i);
        if (c[i] != c[i - 1]) ++ans;
    }
    while (m--) {
        scanf("%d", &f);
        if (f == 2) printf("%d\n", ans);
        else {
            scanf("%d%d", &a, &b);
            if (a == b) continue;
            if (h[fa[a]].size() > h[fa[b]].size())
                swap(fa[a], fa[b]);
            a = fa[a]; b = fa[b];
            foreach(i, h[a]) {
                if (c[*i - 1] == b) --ans;
                if (c[*i + 1] == b) --ans;
                h[b].insert(*i);
            }
            foreach(i, h[a]) c[*i] = b;
            h[a].clear();
        }
    }
    return 0;
}

1483: [HNOI2009]梦幻布丁

Description

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

Input

第一行给出N,M表示布丁的个数和好友的操作次数. 第二行N个数A1,A2…An表示第i个布丁的颜色从第三行起有M行,对于每个操作,若第一个数字是1表示要对颜色进行改变，其后的两个整数X,Y表示将所有颜色为X的变为Y，X可能等于Y. 若第一个数字为2表示要进行询问当前有多少段颜色，这时你应该输出一个整数. 0

Output

针对第二类操作即询问，依次输出当前有多少段颜色.

Sample Input

4 3
1 2 2 1
2
1 2 1
2

Sample Output

3
1