bzoj 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。

n,m<=1000000

Output

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

Sample Input

4 3
1 2 2 1
2
1 2 1
2

Sample Output

3
1

 

思路：

对每个颜色建一条链表，先预处理出开始的序列的ans，对于每次变色操作我们可以对这两种颜色的链表进行启发式合并同时计算对ans的影响.

 

实现代码：

#include<bits/stdc++.h>
using namespace std;
 
const int M = 1e6+10;
int a[M],head[M],next[M],siz[M],ans,col[M];
 
void Merge(int x,int y){
    for(int i = head[x];i;i=next[i]){
        if(a[i-1] == y) ans--;
        if(a[i+1] == y) ans--;
    }
    for(int i = head[x];i;i=next[i]){
        a[i] = y;
        if(!next[i]){
            next[i] = head[y]; head[y] = head[x];
            break;
        }
    }
    siz[y] += siz[x]; head[x] = siz[x] = 0;
}
 
int main()
{
    int n,m,op,x,y;
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= n;i ++){
        scanf("%d",&a[i]);
        if(a[i] != a[i-1]) ans ++;
        col[a[i]] = a[i];
        next[i] = head[a[i]]; head[a[i]] = i;
        siz[a[i]]++;
    }
    for(int i = 1;i <= m;i ++){
        scanf("%d",&op);
        if(op == 1){
            scanf("%d%d",&x,&y);
            if(siz[col[x]] > siz[col[y]]) swap(col[x],col[y]);
            x = col[x]; y = col[y];
            if(x ==y||siz[x] == 0) continue;
            Merge(x,y);
        }
        else
            printf("%d\n",ans);
    }
}