P3939 数颜色 —— 主席树简单变形

题面

数颜色

思路

因为这道题要求区间权值出现次数，很自然的就想到主席树；

这样操作一就解决了；

因为主席树维护的是一个前缀和一样的东西；

因此对于操作二来说，我们只需要修改$x_j$位置上的值即可；

因为这道题卡常，因此要用快读快写；

Code

#include <iostream>
#include <cstdio>
#include <unordered_map>
#include <vector>
#include <algorithm>

using namespace std;

typedef long long ll;

const int N = 3e6 + 10;

struct Tree{
    int lc,rc;
    int sum;//该颜色出现的次数
}tr[N<<5];

int root[N],cnt,a[N],n,m;
vector<int> ve;
int _get(int x){
    return 1 + (lower_bound(ve.begin(),ve.end(),x) - ve.begin());
}
int build(int l,int r){
    int p = ++cnt;
    if(l == r) return p;
    int mid = (l+r) >> 1;
    tr[p].lc = build(l,mid);
    tr[p].rc = build(mid+1,r);
    return p;
}
int update(int p,int l,int r,int val,int k){
    int q = ++cnt;
    tr[q] = tr[p];
    //tr[q].sum = tr[p].sum + k;
    if(l == r){
        tr[q].sum = tr[p].sum + k;
        return q;
    }
    int mid = (l+r) >> 1;
    if(val <= mid){
        tr[q].lc = update(tr[p].lc,l,mid,val,k);
    }else tr[q].rc = update(tr[p].rc,mid+1,r,val,k);
    return q;
}
//查询颜色val在两个版本之间出现的次数
int query(int p,int q,int l,int r,int val){
    if(l == r){
        return tr[q].sum - tr[p].sum;
    }
    int mid = (l+r) >> 1;
    if(val <= mid) return query(tr[p].lc,tr[q].lc,l,mid,val);
    else return query(tr[p].rc,tr[q].rc,mid+1,r,val);
}
int read(){
    int x=0;bool f=0;char c=getchar();
    while (c<'0'||c>'9'){if (c=='-')f=1;c=getchar();}
    while (c>='0'&&c<='9'){x=(x<<1)+(x<<3)+(c^48);c=getchar();}
    return f?-x:x;
}
inline void write(int x){
    if(x<0) putchar('-'),x=-x;
    if(x>9) write(x/10);
    putchar(x%10+'0');
}
unordered_map<int,bool> um;
void solve(){
    n = read(),m = read();
    for(int i=1;i<=n;++i){
        a[i] = read();
        um[a[i]] = 1;
        ve.push_back(a[i]);
    }
    sort(ve.begin(),ve.end());
    ve.erase(unique(ve.begin(),ve.end()),ve.end());
    root[0] = build(1,ve.size());
    for(int i=1,color;i<=n;++i){
        color = _get(a[i]);
        root[i] = update(root[i-1],1,ve.size(),color,1);
    }
    int op,x,y,z;
    while(m--){
        op = read();
        if(op == 1){
            x = read(),y = read(),z = read();
            if(!um.count(z)){
                //如果z根本没出现过
                write(0);
            }
            else write(query(root[x-1],root[y],1,ve.size(),_get(z)));
            putchar('\n');
        }else{
            x = read();
            //把颜色x和x+1互换
            //因为前缀和的原因，因此我们只需要更新版本x即可
            root[x] = update(root[x],1,ve.size(),_get(a[x]),-1);
            root[x] = update(root[x],1,ve.size(),_get(a[x+1]),1);
            swap(a[x],a[x+1]);
        }
    }
}

int main(){
    //std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    solve();
    return 0;
}