题意:
有两个操作
1、将所有的x改成y
2、求序列的颜色段 如1 2 2 1 ans=2(m<=1000000)
首先,可以求出初始序列的ans,然后每次修改颜色相当于合并两个颜色,启发式合并。
启发式合并就是将sz小(s1)的暴力加入sz大的(s2),因为sz[s1]至少翻倍,至多翻倍至totsz,所以复杂度是O(logn)
#include<cstdio>
#include<algorithm>
#define N 1001001
using namespace std;
int num=0,n,m,ans=0;
int a[N],head[N],next[N],sz[N],f[N];
void insert(int k){
num++;
int t=head[k];
head[k]=num;
next[num]=t;
}
void merge(int x,int y){//change x into y
if(x==y) return;
if(sz[f[x]]>sz[f[y]]) swap(f[x],f[y]);
x=f[x];y=f[y];
int t;
if(head[x]==0) return;
for(t=head[x];t;t=next[t]){
if(a[t+1]==y) ans--;
if(a[t-1]==y) ans--;
}
for(t=head[x];t;t=next[t]) a[t]=y;
for(t=head[y];next[t];t=next[t]);
next[t]=head[x];sz[y]+=sz[x];sz[x]=0;
head[x]=0;
}
int main(){
freopen("in.txt","r",stdin);
freopen("out.txt","w",stdout);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
if(a[i]!=a[i-1]) ans++;
insert(a[i]);
sz[a[i]]++;f[a[i]]=a[i];
}
for(int i=1;i<=m;i++){
int d,x,y;
scanf("%d",&d);
if(d==1) scanf("%d%d",&x,&y),merge(x,y);
else printf("%d\n",ans);
}
return 0;
}