bzoj2120 数颜色

Posted 2020-06-08

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