P3939 数颜色 线段树动态开点

P3939 数颜色 线段树动态开点

luogu P3939

水。直接对每种颜色开个权值线段树即可，注意动态开点。

#include 
#include 
#define MAXN 300003
#define MAXM MAXN*30
#define mid ((l+r)>>1)
inline void myswap(int &a, int &b){
    int t=a;a=b;b=t;
}
int tre[MAXM],lson[MAXM],rson[MAXM],tot;
void change(int &cur, int l, int r, int p, int val){
    if(cur==0) cur=++tot;
    tre[cur]+=val;
    if(l==r) return;
    if(p<=mid) change(lson[cur], l, mid, p, val);
    if(mid'9') ch=getchar();
    while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
    return s;
}
int main(){
    n=read(),m=read();
    for(int i=1;i<=n;++i){
        col[i]=read();
        change(rot[col[i]], 1, n, i, 1);
    }
    while(m--){
        int opt;
        opt=read();
        if(opt==1){
            int l=read(),r=read(),c=read();
            printf("%d\n", query(rot[c], 1, n, l, r));
        }else{
            int x=read();
            change(rot[col[x]], 1, n, x, -1);
            change(rot[col[x+1]], 1, n, x+1, -1);
            change(rot[col[x]], 1, n, x+1, 1);
            change(rot[col[x+1]], 1, n, x, 1);
            myswap(col[x], col[x+1]);
        }
    }
    return 0;
}