BZOJ 1483: [HNOI2009]梦幻布丁( 链表 + 启发式合并 )
把相同颜色的串成一个链表, 然后每次A操作就启发式合并, 然后计算对答案的影响.
------------------------------------------------------------------------
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn = 1000009;
int N, M, ans, H;
int c[maxn], C, sz[maxn], Id[maxn];
struct Link {
int p;
Link* next;
} *head[maxn], *last[maxn], pool[maxn], *pt = pool;
void Add(int c, int p) {
if(!head[c])
last[c] = pt;
pt->p = p;
pt->next = head[c];
head[c] = pt++;
}
void Init() {
memset(sz, 0, sizeof sz);
C = 0; ans = 1;
scanf("%d%d", &N, &M);
for(int i = 0; i < N; i++) {
scanf("%d", c + i);
Add(Id[c[i]] = c[i], i);
sz[c[i]]++;
if(i && c[i] != c[i - 1]) ans++;
}
}
void Change(int oc, int nc) {
for(Link* r = head[oc]; r; r = r->next) {
if(r->p > 0 && c[r->p - 1] == nc) ans--;
if(r->p + 1 < N && c[r->p + 1] == nc) ans--;
}
for(Link* r = head[oc]; r; r = r->next) c[r->p] = nc;
last[nc]->next = head[oc];
last[nc] = last[oc];
}
int main() {
Init();
while(M--) {
int opt;
scanf("%d", &opt);
if(opt == 2)
printf("%d\n", ans);
else {
int oc, nc;
scanf("%d%d", &oc, &nc);
if(sz[Id[oc]] > sz[Id[nc]])
swap(Id[oc], Id[nc]);
oc = Id[oc]; nc = Id[nc];
if(!sz[oc] || oc == nc) continue;
sz[nc] += sz[oc]; sz[oc] = 0;
Change(oc, nc);
}
}
return 0;
}
------------------------------------------------------------------------
1483: [HNOI2009]梦幻布丁
Time Limit: 10 Sec Memory Limit: 64 MBSubmit: 1686 Solved: 713
[Submit][Status][Discuss]
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
1 2 2 1
2
1 2 1
2
Sample Output
3
1
1
HINT
Source
