主席树(静态区间第k小)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<ctime>
#include<queue>
using namespace std;
const int M=200005;
int Cnt,n,m,l,r,k,q,ans,sum[M<<5],root[M],lc[M<<5],rc[M<<5],a[M],b[M],p;
void build(int &t,int l,int r){
    t=++Cnt;
    if(l==r){
        return;
	}
    int mid=(l+r)>>1;
    build(lc[t],l,mid);
    build(rc[t],mid+1,r);
}
int modify(int o,int l,int r){
    int oo=++Cnt;
    lc[oo]=lc[o];
	rc[oo]=rc[o];
	sum[oo]=sum[o]+1;
    if(l==r){
        return oo;
	}
    int mid=(l+r)>>1;
    if(p<=mid){
		lc[oo]=modify(lc[oo],l,mid);
	}
    else{
		rc[oo]=modify(rc[oo],mid+1,r);
	}
    return oo;
}
int query(int u,int v,int l,int r,int k){
    int ans,mid=((l+r)>>1),x=sum[lc[v]]-sum[lc[u]];
    if(l==r){
        return l;
	}
    if(x>=k){
		ans=query(lc[u],lc[v],l,mid,k);
	}
    else{
		ans=query(rc[u],rc[v],mid+1,r,k-x);
	}
    return ans;
}
int main(){
    scanf("%d%d", &n, &m);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
		b[i]=a[i];
	}
    sort(b+1, b+n+1);
    q=unique(b+1,b+n+1)-b-1;
    build(root[0],1,q);
    for(int i=1;i<=n;i++){
        p=lower_bound(b+1,b+q+1,a[i])-b;
        root[i]=modify(root[i-1],1,q);
    }
    while(m--){
        scanf("%d%d%d",&l,&r,&k);
        ans=query(root[l-1],root[r],1,q,k);
        printf("%d\n", b[ans]);
    }
    return 0;
}
posted @ 2019-07-16 13:39  prestige  阅读(198)  评论(0编辑  收藏  举报