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;
}