题意: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;
}