[HNOI2009] 梦幻布丁

XLV.[HNOI2009] 梦幻布丁

线段树合并是非常显然的，但是这里我们偏不用。这里我们使用的是启发式合并——虽然这仍然非常显然。

可以使用链表做到 \(O(n\log n)\) 但是我太懒了因此直接暴力用 set 做了，是 \(O(n\log^2n)\) 的不过一样也能过。

附：set 直接 swap 是 \(O(1)\) 的哦！

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1000000;
int n,m,res;
set<pair<int,int> >s[1001000];
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1,x;i<=n;i++){
		scanf("%d",&x);
		int l=i,r=i;
		if(!s[x].empty()&&s[x].rbegin()->second==i-1)l=s[x].rbegin()->first,s[x].erase(--s[x].end());
		s[x].emplace(l,r);
	}
	for(int i=1;i<=N;i++)res+=s[i].size();
	for(int i=1,tp,x,y;i<=m;i++){
		scanf("%d",&tp);
		if(tp==1){
			scanf("%d%d",&x,&y);
			if(x==y||s[x].empty())continue;
			if(s[y].empty()){swap(s[x],s[y]);continue;}
			bool swt=true;
			if(s[x].size()<s[y].size())swt=false,swap(x,y);
			res-=s[x].size(),res-=s[y].size();
			for(auto i:s[y]){
				int l=i.first,r=i.second;
				auto it=s[x].lower_bound(i);
				if(it!=s[x].end()&&it->first==r+1)r=it->second,it=s[x].erase(it);
				if(it!=s[x].begin()&&(--it)->second==l-1)l=it->first,s[x].erase(it);
				s[x].emplace(l,r);
			}
			res+=s[x].size(),s[y].clear();
			if(swt)swap(s[x],s[y]);			
		}else printf("%d\n",res);
	}
	return 0;
}