正题

题目链接:https://www.luogu.com.cn/problem/P3201

---

题目大意

开始有$n$个布丁，第$i$个是$c_i$颜色的。
每次有操作

1. 将所有颜色为$x$的布丁变为颜色$y$的。
2. 询问有多少个布丁颜色段。

---

解题思路

对于每次修改，我们可以考虑启发式合并，将小的合并到大的上面，我们可以用类似于链表的结构维护每个颜色的所有布丁。修改后直接将数量少的合并到数量大的上面即可。

时间复杂度$O(n\log n)$

---

$code$

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e6+10;
int n,m,ans,fa[N],a[N],siz[N];
int ls[N],st[N],next[N];
void Merge(int x,int y){for(int i=ls[x];i;i=next[i])ans-=(a[i-1]==y)+(a[i+1]==y);for(int i=ls[x];i;i=next[i])a[i]=y;next[st[x]]=ls[y];ls[y]=ls[x];siz[y]+=siz[x];ls[x]=siz[x]=st[x]=0;
}
int main()
{scanf("%d%d",&n,&m);for(int i=1;i<=n;i++){scanf("%d",&a[i]);fa[a[i]]=a[i];ans+=a[i]!=a[i-1];if(!st[a[i]])st[a[i]]=i;siz[a[i]]++;next[i]=ls[a[i]];ls[a[i]]=i;}for(int i=1;i<=m;i++){int op,x,y;scanf("%d",&op);if(op==1){scanf("%d%d",&x,&y);if(x==y)continue;if(siz[fa[x]]>siz[fa[y]])swap(fa[x],fa[y]);if(!siz[fa[x]])continue;Merge(fa[x],fa[y]);}else printf("%d\n",ans);}return 0;
}