题目大意:可以把一种颜色完全染为另一种颜色,问颜色段数。
我用了线段树合并。对每种颜色开一棵线段树每一段记下左边和右边是否有这种颜色,就可以维护了。
然而无限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;
}
}
}