[FROM LUOGU]P3201 [HNOI2009]梦幻布丁

最新推荐文章于 2022-09-11 19:43:47 发布

Park Corsa

最新推荐文章于 2022-09-11 19:43:47 发布

阅读量113

点赞数

分类专栏： ---数据结构--- # 启发式合并文章标签： LUOGU

本文链接：https://blog.csdn.net/hzq_oi/article/details/102232990

版权

---数据结构--- 同时被 2 个专栏收录

28 篇文章 0 订阅

订阅专栏

启发式合并

1 篇文章 0 订阅

订阅专栏

P3201 [HNOI2009]梦幻布丁

传送门

SOL
$O(n^2)$ 做法：
先按颜色将每一个点用邻接表存储，记录每条链的始末和 $s i z$
由于有些颜色可能后来被合并掉了，修改时可能找不到目标，因此用 $f a [c [i]]$ 记录一下对应的链
对于每一个修改操作直接暴力修改答案，然后暴力重新染色，然后将一条链接到另一条的后面
$O (n (l o g n))$ 做法：
在前面的暴力基础上用启发式合并优化即可
短的链合并到长的链上，显然一条链每次合并之后大小至少为原来的2倍，于是最坏情况也只有 $l o g n$ 次

代码：

#include<iostream>
#include<cstdio>
using namespace std;
#define re register
inline char nc(){
    static char buf[10000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,10000,stdin),p1==p2)?EOF:*p1++;
}
inline int rd(){
	int data=0;static char ch=0;
	while(!isdigit(ch=nc()));data=ch^48;
	while(isdigit(ch=nc()))data=(data<<3)+(data<<1)+(ch^48);
	return data;
}
const int N=1e5+5,M=1e6+5;
int n,m,ans,c[N],cur[M],nxt[N],st[M],fa[M],siz[M];
inline void merge(int x,int y){
	for(int re i=cur[x];i;i=nxt[i])ans-=(c[i-1]==y)+(c[i+1]==y);
	for(int re i=cur[x];i;i=nxt[i])c[i]=y;
	nxt[st[x]]=cur[y],cur[y]=cur[x],siz[y]+=siz[x],cur[x]=st[x]=siz[x]=0;
}
inline void write(int x){if(x>9)write(x/10);putchar((x%10)^48);}
int main(){
	n=rd()+1,m=rd();
	for(int re i=1;i^n;++i){
		ans+=(c[i]=rd())!=c[i-1];
		if(!cur[c[i]])st[c[i]]=i,fa[c[i]]=c[i];
		nxt[i]=cur[c[i]],cur[c[i]]=i,++siz[c[i]];
	}
	while(m--){
		int re op=rd(),x,y;
		if(op==2)write(ans),cout<<"\n";
		else{
			if((x=rd())==(y=rd()))continue;
			if(siz[fa[x]]>siz[fa[y]])swap(fa[x],fa[y]);
			if(!siz[fa[x]])continue;
			merge(fa[x],fa[y]);
		}
	}return 0;
}