Description
墨墨购买了一套N支彩色画笔（其中有些颜色可能相同），摆成一排，你需要回答墨墨的提问。墨墨会像你发布如下指令： 1、 Q L R代表询问你从第L支画笔到第R支画笔中共有几种不同颜色的画笔。 2、 R P Col 把第P支画笔替换为颜色Col。为了满足墨墨的要求，你知道你需要干什么了吗？
对于100%的数据，N≤10000，M≤10000，修改操作不多于1000次，所有的输入数据中出现的所有整数均大于等于1且不超过10^6。
Input
第1行两个整数N，M，分别代表初始画笔的数量以及墨墨会做的事情的个数。第2行N个整数，分别代表初始画笔排中第i支画笔的颜色。第3行到第2+M行，每行分别代表墨墨会做的一件事情，格式见题干部分。
Output
对于每一个Query的询问，你需要在对应的行中给出一个数字，代表第L支画笔到第R支画笔中共有几种不同颜色的画笔。
treap维护每种颜色出现的所有位置
zkw线段树套treap维护区间中每种颜色的上次出现位置
查询[l,r]时统计区间[l,r]内上次出现位置<r的值的个数
#include<cstdio> #include<cstdlib> #define N 4000000 int ch[N][2]; int sz[N],a[N]; int v[N]; int p=1,V; inline void upd(int x){ if(x)sz[x]=sz[ch[x][0]]+sz[ch[x][1]]+1; } struct bst{ int root; bst(){ root=0; } void insert(int x){ v[p]=V=x; sz[p]=1; a[p]=rand(); if(root)_insert(root); else root=p++; } void _insert(int&x){ bool d=V>v[x]; int&c=ch[x][d]; if(c)_insert(c); else c=p++; if(a[ch[x][d]]>a[x])rot(x,d); upd(x); } void rot(int&w,int c){ int u=ch[w][c]; ch[w][c]=ch[u][c^1]; ch[u][c^1]=w; upd(w); w=u; upd(u); } int lss(int x){ int w=root,ans=0; while(w){ if(v[w]<x)ans+=sz[ch[w][0]]+1; w=ch[w][v[w]<x]; } return ans; } void del(int&w,int V){ if(!w)return; if(v[w]==V)del(w); else del(ch[w][v[w]<V],V); upd(w); } void del(int&w){ if(ch[w][0]|ch[w][1]){ bool d=a[ch[w][0]]<a[ch[w][1]]; int u=ch[w][d]; rot(w,d); del(ch[w][d^1]); upd(u); }else w=0; } int prv(int x){ int w=root,a=-1; while(w){ if(x>v[w])a=v[w]; w=ch[w][x>v[w]]; } return a; } int nxt(int x){ int w=root,a=-1; while(w){ if(x<v[w])a=v[w]; w=ch[w][x>=v[w]]; } return a; } }; bst cp[1000001]; bst tr[32768]; int col[10001]; void tr_ins(int i,int x){ for(i+=16384;i>0;i>>=1)tr[i].insert(x); } void tr_del(int i,int x){ for(i+=16384;i>0;i>>=1)tr[i].del(tr[i].root,x); } int tr_q(int l,int r){ int ans=0,lp=l; for(l+=16383,r+=16385;l^r^1;l>>=1,r>>=1){ if(~l&1)ans+=tr[l^1].lss(lp); if(r&1)ans+=tr[r^1].lss(lp); } return ans; } int main(){ int n,m,l,r; char op; for(int i=0;i<=100000;i++){ cp[i].insert(0); } scanf("%d%d",&n,&m); for(int i=1;i<=n;i++){ scanf("%d",col+i); cp[col[i]].insert(i); tr_ins(i,cp[col[i]].prv(i)); } while(m--){ scanf(" %c %d%d",&op,&l,&r); if(op=='Q'){ printf("%d\n",tr_q(l,r)); }else if(r!=col[l]){ int vl=col[l]; int nx=cp[vl].nxt(l); int pv=cp[vl].prv(l); int pn=cp[r].prv(l); int nn=cp[r].nxt(l); col[l]=r; tr_del(l,pv); tr_ins(l,pn); if(nx>0){ tr_del(nx,l); tr_ins(nx,pv); } if(nn>0){ tr_del(nn,pn); tr_ins(nn,l); } cp[vl].del(cp[vl].root,l); cp[r].insert(l); } } return 0; }
