poj 2104 K-th Number （主席树模板）

传送门

// by spli#include<cstring>#include<cstdio>#include<algorithm>#include<iostream>using namespace std;const int N=200010;int n,m;struct node{    int id,v;}a[N];int mp[N];int r[N];struct seg{    int ls,rs;    int siz;}t[N*40];int tot;bool cmp(node x,node y){    return x.v<y.v;}void update(int &pos,int L,int R,int v){    t[++tot]=t[pos];pos=tot;    t[pos].siz++;    if(L==R) return;    int mid=(L+R)>>1;    if(v<=mid) update(t[pos].ls,L,mid,v);    else update(t[pos].rs,mid+1,R,v);}int query(int ll,int rr,int L,int R,int k){    if(L==R) return L;    int s=t[t[rr].ls].siz-t[t[ll].ls].siz;    int mid=(L+R)>>1;    if(k<=s) return query(t[ll].ls,t[rr].ls,L,mid,k);    else return query(t[ll].rs,t[rr].rs,mid+1,R,k-s);}int main(){    scanf("%d%d",&n,&m);    for(int i=1;i<=n;++i) scanf("%d",&a[i].v),a[i].id=i;    sort(a+1,a+1+n,cmp);    for(int i=1;i<=n;++i) mp[a[i].id]=i;//保证离散化后各不相同,不能使用lower_bound     for(int i=1;i<=n;++i) r[i]=r[i-1],update(r[i],1,n,mp[i]);    int x,y,k;    while(m--){        scanf("%d%d%d",&x,&y,&k);        printf("%d\n",a[query(r[x-1],r[y],1,n,k)].v);    }    return 0;}