主席树板子题区间第k小

https://www.luogu.org/problem/P3834

#include<bits/stdc++.h>
using namespace std;
typedef long long ll ;
const int maxn=2e5+7;
int n,m,cnt,root[maxn],a[maxn],x,y,k;
struct node
{
    int l,r,sum;
} T[maxn*25];
int b[maxn];
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)/2;
    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)/2;
    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(void)
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
        b[i]=a[i];
    }
    sort(b+1,b+1+n);
    int cntb=unique(b+1,b+1+n)-1-b;
    for(int i=1; i<=n; i++)
    {
        int pos=lower_bound(b+1,b+1+cntb,a[i])-b;
        update(1,n,root[i],root[i-1],pos);
    }
    for(int i=1; i<=m; i++)
    {
        scanf("%d%d%d",&x,&y,&k);
        printf("%d\n",b[query(1,n,root[x-1],root[y],k)]);
    }
    return 0;
}

View Code