主席树模板 更新

#include <bits/stdc++.h>
const int maxn=2*1e5+100;
using namespace std;

int n,m,a[maxn],root[maxn],x,y,k,cnt = 0;
struct node{
    int l,r,sum;
}T[maxn*40];
vector <int> v; 

int getid(int x){
    return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}

void update(int l,int r,int &x,int y,int pos){
    T[++cnt] = T[y],T[cnt].sum++,x = cnt;
    if (l == r) return;
    int mid = (l+r)>>1;
    if (mid >= pos) update(l,mid,T[x].l,T[y].l,pos);
    else update(mid+1,r,T[x].r,T[y].r,pos);
}

int query(int l,int r,int x,int y,int k){
    if (l == r) return l;
    int mid=(l+r)>>1;
    int sum = T[T[y].l].sum - T[T[x].l].sum;
    if (sum >= k ) return query(l,mid,T[x].l,T[y].l,k);
    else return query(mid+1,r,T[x].r,T[y].r,k-sum);
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin >> n >> m;

    for (int i=1;i<=n;i++) cin >> a[i],v.push_back(a[i]);
    sort(v.begin(),v.end()),v.erase(unique(v.begin(),v.end()),v.end());

    for (int i=1;i<=n;i++) update(1,n,root[i],root[i-1],getid(a[i]));
    for (int i=1;i<=m;i++){
        cin >> x >> y >> k;
        cout << v[query(1,n,root[x-1],root[y],k)-1] << endl;
    }

    return 0;
}

模板题：

P3834 【模板】可持久化线段树 1（主席树）