[HNOI2009] 梦幻布丁

标签： 链表

---

题解

可以直接用链表启发式合并做。

合并的细节处理稍微有点麻烦。

假如需要变成另一种颜色的那个颜色的个数更多，那么就肯定不能直接合。

维护一个color数组代表，每个颜色在数组中存的位置。

这样交换color数组之后可以直接合并。

Code

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<iostream>

#include<algorithm>

#include<cmath>

#include<queue>

#include<stack>

#include<set>

#include<map>

using namespace std;

#define ll long long

#define REP(i,a,b) for(int i=(a),_end_=(b);i<=_end_;i++)

#define DREP(i,a,b) for(int i=(a),_end_=(b);i>=_end_;i--)

#define EREP(i,a) for(int i=start[(a)];i;i=e[i].next)

inline int read()

{

	int sum=0,p=1;char ch=getchar();

	while(!(('0'<=ch && ch<='9') || ch=='-'))ch=getchar();

	if(ch=='-')p=-1,ch=getchar();

	while('0'<=ch && ch<='9')sum=sum*10+ch-48,ch=getchar();

	return sum*p;

}

const int maxn=1e5+20;

int n,m,nxt[maxn],sz[maxn*10],col[maxn],start[maxn*10],color[maxn*10],fro[maxn*10];

int ans=0;

inline void init()

{

	n=read();

	m=read();

	REP(i,1,n)

	{

		col[i]=read();

		int x=col[i];

		if(col[i-1]!=x)ans++;

		if(!start[x])fro[x]=i;

		nxt[i]=start[x];

		start[x]=i;

		sz[x]++;

		color[x]=x;

	}

}

inline void doing()

{

	REP(i,1,m)

	{

		if(read()==2)printf("%d\n",ans);

		else

		{

			register int y=read(),x=read();

			if(x!=y)

			{

				if(sz[color[x]]<sz[color[y]])swap(color[x],color[y]);

				x=color[x];y=color[y];

				if(!sz[y])continue;

				sz[x]+=sz[y];sz[y]=0;

				for(int u=start[y];u;u=nxt[u])ans-=(col[u-1]==x)+(col[u+1]==x);

				for(int u=start[y];u;u=nxt[u])col[u]=x;

				nxt[fro[y]]=start[x];

				start[x]=start[y];

				start[y]=0;

				sz[y]=0;

				start[y]=0;

			}

		}

	}

}

int main()

{

	init();

	doing();

	return 0;

}