正題

題目鏈接:https://www.luogu.com.cn/problem/P3201

---

題目大意

開始有$n$個布丁，第$i$個是$c_i$顏色的。
每次有操作

將所有顏色為$x$的布丁變?yōu)轭伾?span id="ozvdkddzhkzd" class="katex--inline">$y$的。

詢問有多少個布丁顏色段。

---

解題思路

對于每次修改，我們可以考慮啟發(fā)式合并，將小的合并到大的上面，我們可以用類似于鏈表的結(jié)構(gòu)維護每個顏色的所有布丁。修改后直接將數(shù)量少的合并到數(shù)量大的上面即可。

時間復(fù)雜度$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; }

總結(jié)

以上是生活随笔為你收集整理的P3201-[HNOI2009]梦幻布丁【启发式合并,链表】的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。