POJ2104——K-th Number (主席树模板)
传送门
主席树模板,不想多说了
具体的神仙讲解
#include<algorithm>
#include<cmath>
#include<iostream>
#include<stdio.h>
#include<cstdio>
#include<cstdlib>
#include<string.h>
#include<cstring>
using namespace std;
#define mid ((l+r)>>1)
inline int read(){
char ch=getchar();
int res=0,f=1;
while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)) res=(res<<3)+(res<<1)+(ch^48),ch=getchar();
return res*f;
}
int n,m,q,tot,num[200005],a[200005],sum[4000005],lc[4000005],rc[4000005],root[4000050];
inline void init(){
sort(a+1,a+n+1);
m=unique(a+1,a+1+n)-a-1;
for(int i=1;i<=n;i++)num[i]=lower_bound(a+1,a+1+m,num[i])-a;
}
inline void buildtree(int fa,int &now,int l,int r,int num){
now=++tot;
sum[now]=sum[fa]+1,lc[now]=lc[fa],rc[now]=rc[fa];
if(l==r) return;
if(num<=mid) buildtree(lc[fa],lc[now],l,mid,num);
else buildtree(rc[fa],rc[now],mid+1,r,num);
}
inline int query(int r1,int r2,int l,int r,int k){
if(l==r)return l;
int num=sum[lc[r1]]-sum[lc[r2]];
if(num>=k) return query(lc[r1],lc[r2],l,mid,k);
else return query(rc[r1],rc[r2],mid+1,r,k-num);
}
int main(){
n=read(),q=read();
for(int i=1;i<=n;i++) num[i]=a[i]=read();
init();
for(int i=1;i<=n;i++)buildtree(root[i-1],root[i],1,m,num[i]);
while(q--){
int l=read(),r=read(),k=read();
cout<<a[query(root[r],root[l-1],1,m,k)]<<'\n';
}
}