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

链接

点击跳转

代码

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define iinf 0x3f3f3f3f
#define linf (1ll<<60)
#define eps 1e-8
#define maxn 1000010
#define maxe 1000010
#define cl(x) memset(x,0,sizeof(x))
#define rep(_,__) for(_=1;_<=(__);_++)
#define em(x) emplace(x)
#define emb(x) emplace_back(x)
#define emf(x) emplace_front(x)
#define fi first
#define se second
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll read(ll x=0)
{
    ll c, f(1);
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
    for(;isdigit(c);c=getchar())x=x*10+c-0x30;
    return f*x;
}
struct PersistenceSegmentTree
{
    int root[maxn<<5], rtot, tot, L[maxn<<5], R[maxn<<5], ch[maxn<<5][2], v[maxn<<5];
    int copy(int o)
    {
        tot++;
        L[tot]=L[o], R[tot]=R[o], ch[tot][0]=ch[o][0], ch[tot][1]=ch[o][1], v[tot]=v[o];
        return tot;
    }
    int New()
    {
        tot++;
        L[tot]=R[tot]=ch[tot][0]=ch[tot][1]=v[tot]=0;
        return tot;
    }
    void build(int o, int l, int r)
    {
        int mid(l+r>>1);
        L[o]=l, R[o]=r;
        if(l==r)return;
        ch[o][0]=New(), ch[o][1]=New();
        build(ch[o][0],l,mid);
        build(ch[o][1],mid+1,r);
    }
    void init(int l, int r)
    {
        rtot=0, tot=0;
        build(root[0]=New(),l,r);
    }
    int insert(int o, int pos, int add)
    {
        int x=copy(o), mid=L[o]+R[o]>>1;
        v[x]+=add;
        if(L[o]==R[o])return x;
        if(pos<=mid)ch[x][0]=insert(ch[o][0],pos,add);
        else ch[x][1]=insert(ch[o][1],pos,add);
        return x;
    }
    void ins(int pos, int add=1)
    {
        root[rtot+1]=insert(root[rtot],pos,add);
        rtot++;
    }
    int qsum(int o1, int o2, int l, int r)
    {
        int ans=0, mid=L[o1]+R[o1]>>1;
        if(l<=L[o1] and r>=R[o1])return v[o2]-v[o1];
        if(l<=mid)ans+=qsum(ch[o1][0],ch[o2][0],l,r);
        if(r>mid)ans+=qsum(ch[o1][1],ch[o2][1],l,r);
        return ans;
    }
    int kth(int o1, int o2, int k)
    {
        if(L[o1]==R[o1])return L[o1];
        int t=v[ch[o2][0]]-v[ch[o1][0]];
        if(t>=k)return kth(ch[o1][0],ch[o2][0],k);
        else return kth(ch[o1][1],ch[o2][1],k-t);
    }
}zx;
struct Lisan
{
    int tmp[maxn], tot;
    void clear(){tot=0;}
    void insert(int x){tmp[++tot]=x;}
    void run()
    {
        sort(tmp+1,tmp+tot+1);
        tot=unique(tmp+1,tmp+tot+1)-tmp-1;
    }
    void lisan(int *a, int len)
    {
        for(int i=1;i<=len;i++)a[i]=lower_bound(tmp+1,tmp+tot+1,a[i])-tmp;
    }
    int lisan(int x)
    {
        return lower_bound(tmp+1,tmp+tot+1,x)-tmp;
    }
}ls;
int n, a[maxn], m;
int main()
{
    int i, l, r, k;
    n=read(), m=read();
    rep(i,n)a[i]=read(), ls.insert(a[i]);
    ls.run();
    zx.init(1,ls.tot);
    rep(i,n)zx.ins(ls.lisan(a[i]));
    while(m--)
    {
        l=read(), r=read(), k=read();
        printf("%d\n",ls.tmp[zx.kth(zx.root[l-1],zx.root[r],k)]);
    }
    return 0;
}