bzoj 1483: [HNOI2009]梦幻布丁 //启发式合并

题意：N个布丁摆成一行,进行M次操作.每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色.(N,M 1e6)

思路：启发式合并

用vector存每种颜色的下标，然后对于每一次合并，将size小的vector合并到size大的vector上，这样每一次操作,小的vector的size都至少翻倍，这样复杂度就是nlogn。

但是这题有可能需要大的size合并到小的size上，这时候可以交换一下颜色，每次调用的时候去调用当前赋予的颜色。

/*   (o O o)
 *   Author : Rshs
 *   Data : 2019-08-28-20.40
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const LL mod = 1e9+7;
const int MX = 1e6+5;
int a[MX],f[MX];vector<int>v[MX];
int main(){
    int n,m;cin>>n>>m;
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        f[a[i]]=a[i];v[a[i]].push_back(i);
    }
    int ans=1;
    for(int i=2;i<=n;i++) if(a[i]!=a[i-1])ans++;
    for(;m;m--){
        int op;scanf("%d",&op);
        if(op==1){
            int s,t;
            scanf("%d%d",&s,&t);
            if(s==t)continue;
            int sz1=SZ(v[f[s]]),sz2=SZ(v[f[t]]);
            if(sz1>sz2)swap(f[s],f[t]); //互换颜色
            for(int i=0;i<SZ(v[f[s]]);i++){
                v[f[t]].push_back(v[f[s]][i]);
                if(f[t]==a[v[f[s]][i]-1]) ans--;
                if(f[t]==a[v[f[s]][i]+1]) ans--;
            }
            for(int i=0;i<SZ(v[f[s]]);i++) a[v[f[s]][i]]=f[t];
            v[f[s]].clear();
        }
        else {
            cout<<ans<<'\n';
        }
    }
    return 0;
}