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;}}}