从队友那偷来的主席树模板(静态区间第k小)

 

#include<bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef pair<int,int> PII;
const int MAXN = 1e6+10;
const double EPS = 1e-12;

int n,m,cnt;
int sum[MAXN<<5],rt[MAXN<<5],Ls[MAXN<<5],Rs[MAXN<<5];
int a[MAXN],b[MAXN];

int Build(int l,int r){
    int root=++cnt;
    if(l==r)return root;
    int mid=(l+r)/2;
    Ls[root]=Build(l,mid);
    Rs[root]=Build(mid+1,r);
    return root;
}

int Update(int k,int l,int r,int root){
    int id=++cnt;
    Ls[id]=Ls[root];Rs[id]=Rs[root];sum[id]=sum[root]+1;
    if(l==r)return id;
    int mid=(l+r)/2;
    if(k<=mid)Ls[id]=Update(k,l,mid,Ls[id]);
    if(k>mid)Rs[id]=Update(k,mid+1,r,Rs[id]);
    return id;
}

int Query(int k,int u,int v,int l,int r){
    if(l==r)return l;
    int mid=(l+r)/2;
    int x=sum[Ls[v]]-sum[Ls[u]];
    if(k<=x)return Query(k,Ls[u],Ls[v],l,mid);
    if(k>x)return Query(k-x,Rs[u],Rs[v],mid+1,r);
}

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);
    int len=unique(b+1,b+n+1)-b-1;
    rt[0]=Build(1,len);
    for(int i=1;i<=n;i++){
        int t=lower_bound(b+1,b+len+1,a[i])-b;
        rt[i]=Update(t,1,len,rt[i-1]);
    }
    int l,r,k;
    while(m--){
        scanf("%d %d %d",&l,&r,&k);
        cout<<b[Query(k,rt[l-1],rt[r],1,len)]<<endl;
    }
    return 0;
}

 

posted @ 2019-11-12 14:28  Mmasker  阅读(130)  评论(0编辑  收藏  举报