[bzoj1483]梦幻布丁

对于每一个颜色用一个链表存储，并记录下：1.当前某种颜色的真实颜色；2.这种颜色的数量（用于启发式合并的判断）；3.当前答案（即有几段），然后对于每一个操作简单处理一下就行了。

#include<bits/stdc++.h>
using namespace std;
#define N 1000005
int n,m,ans,p,x,y,head[N],tru[N],sum[N],nex[N],last[N],a[N];
void merge(int x,int y){
    for(int i=head[x];i!=-1;i=nex[i])ans-=(a[i-1]==y)+(a[i+1]==y);
    for(int i=head[x];i!=-1;i=nex[i])a[i]=y;
    sum[y]+=sum[x];
    nex[last[y]]=head[x];
    last[y]=last[x];
    head[x]=-1;
    sum[x]=last[x]=0;
}
int main(){
    scanf("%d%d",&n,&m);
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=n;i;i--){
        if (a[i]!=a[i-1])ans++;
        tru[a[i]]=a[i];
        if (head[a[i]]==-1)last[a[i]]=i;
        sum[a[i]]++;
        nex[i]=head[a[i]];
        head[a[i]]=i;
    }
    for(int i=1;i<=m;i++){
        scanf("%d",&p);
        if (p==2)printf("%d\n",ans);
        else{
            scanf("%d%d",&x,&y);
            if (sum[tru[x]]>sum[tru[y]])swap(tru[x],tru[y]);
            if ((x!=y)&&(sum[tru[x]]))merge(tru[x],tru[y]);
        }
    }
}