bzoj 1483: [HNOI2009]梦幻布丁

题目大意：可以把一种颜色完全染为另一种颜色，问颜色段数。

我用了线段树合并。对每种颜色开一棵线段树每一段记下左边和右边是否有这种颜色，就可以维护了。

然而无限RE，拍又拍不出。留坑。

难道要离散化？可是std也没有啊？

向大神求助。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
struct trnode{
	int lc,rc,c;
	bool l,r;
	trnode(){l=r=false;c=0;}
}tr[4000010];int root[100010],trlen=0;
int n,m;
void update(int &x,int l,int r,int k)
{
	if(!x) x=++trlen;
	if(l==r){tr[x].l=tr[x].r=true;tr[x].c=1;return;}
	int mid=(l+r)/2;
	if(k<=mid) update(tr[x].lc,l,mid,k);
	else update(tr[x].rc,mid+1,r,k);
	int lc=tr[x].lc,rc=tr[x].rc;
	tr[x].l=tr[lc].l;tr[x].r=tr[rc].r;
	tr[x].c=tr[lc].c+tr[rc].c;
	if(tr[lc].r&&tr[rc].l) tr[x].c--;
}
void merge(int &x,int x2)
{
	if(x==0){x=x2;return;}
	if(x2==0) return;
	merge(tr[x].lc,tr[x2].lc);
	merge(tr[x].rc,tr[x2].rc);
	int lc=tr[x].lc,rc=tr[x].rc;
	tr[x].l=tr[lc].l;tr[x].r=tr[rc].r;
	tr[x].c=tr[lc].c+tr[rc].c;
	if(tr[lc].r&&tr[rc].l) tr[x].c--;
}
int main()
{
	scanf("%d %d",&n,&m);
	memset(root,0,sizeof(root));
	for(int i=1;i<=n;i++)
	{
		int x;scanf("%d",&x);
		update(root[x],1,n,i);
	}
	int ans=0;
	for(int i=1;i<=100000;i++) ans+=tr[root[i]].c;
	while(m--)
	{
		int tmp;scanf("%d",&tmp);
		if(tmp==2) printf("%d\n",ans);
		else
		{
			int x,y;scanf("%d %d",&x,&y);
			if(x==y) continue;
			ans-=tr[root[x]].c;ans-=tr[root[y]].c;
			merge(root[y],root[x]);
			root[x]=0;ans+=tr[root[y]].c;
		}
	}
}