主席树练习1——P3834 【模板】可持久化线段树 1（主席树）

蒟蒻的垂死挣扎

主席树真可怕。

还是之前学过一些嘛，不至于完全懵逼。重新打一遍模板练练手。

这个解决的是经典的区间第k大问题，开值域线段树维护，做线段树减法即可。

// luogu-judger-enable-o2
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#define RG register
#define N 200100
#define ll long long
#define ld long double
using namespace std;

inline int read(){
  RG int x=0,o=1; RG char ch=getchar();
  while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
  if(ch=='-') o=-1,ch=getchar();
  while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
  return x*o;
}

int len,n,m,a[N],b[N],yyb[N],tot,root[N];
struct mona { int le,ri; int w; } tr[N*80];

inline void Orz(){
    sort(yyb+1,yyb+1+n);
    len=unique(yyb+1,yyb+1+n)-yyb-1;
    for(RG int i=1;i<=n;++i) a[i]=lower_bound(yyb+1,yyb+1+len,a[i])-yyb;
}

inline void Build(RG int l,RG int r,RG int &x,RG int y,RG int w){
    tr[++tot]=tr[y],++tr[tot].w,x=tot;
    if(l==r) return ;
    RG int mid=(l+r)>>1;
    if(w<=mid) Build(l,mid,tr[x].le,tr[y].le,w);
    else Build(mid+1,r,tr[x].ri,tr[y].ri,w);
}

inline int Query(RG int l,RG int r,RG int x,RG int y,RG int k){
    if(l==r) return yyb[l];
    RG int mid=(l+r)>>1;
    RG int sum=tr[tr[y].le].w-tr[tr[x].le].w;
    if(k<=sum) return Query(l,mid,tr[x].le,tr[y].le,k);
    else return Query(mid+1,r,tr[x].ri,tr[y].ri,k-sum);
}

int main(){
    n=read(),m=read();
    for(RG int i=1;i<=n;++i) a[i]=b[i]=yyb[i]=read();
    Orz();
    for(RG int i=1;i<=n;++i) Build(1,n,root[i],root[i-1],a[i]);
    for(RG int i=1;i<=m;++i){
        RG int x=read(),y=read(),z=read();
        printf("%d\n",Query(1,n,root[x-1],root[y],z));
    }
}