可持久化线段树 不修改第k大模板

https://www.luogu.org/problemnew/show/P3834

http://poj.org/problem?id=2104

第一种建一颗空树，进行插入。

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#include<queue>
using namespace std;
typedef long long ll;
const int N=2e5+10;
int ls[N*40],rs[N*40],val[N*40],sum[N*40];
int cnt=0,w[N],q[N],root[N*40];
int build(int l,int r)
{
    int now=++cnt;
    if(l==r) return now;
    int mid=(l+r)>>1;
    ls[now]=build(l,mid);
    rs[now]=build(mid+1,r);
    return now;
}
int update(int x,int l,int r,int old)
{
    int now=++cnt;
    ls[now]=ls[old],rs[now]=rs[old];
    sum[now]=sum[old]+1;
    if(l==r) return now;
    int mid=(l+r)>>1;
    if(x<=mid) ls[now]=update(x,l,mid,ls[old]);
    else rs[now]=update(x,mid+1,r,rs[old]);
    return now;
}
int query(int lrt,int rrt,int l,int r,int k)
{
    if(l==r) return l;
    int t=sum[ls[rrt]]-sum[ls[lrt]];
    int mid=(l+r)>>1;
    if(k<=t) return query(ls[lrt],ls[rrt],l,mid,k);
    else return query(rs[lrt],rs[rrt],mid+1,r,k-t);
}

int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]),q[i]=w[i];
    sort(q+1,q+n+1);
    int len=unique(q+1,q+n+1)-q-1;
    for(int i=1;i<=n;i++) w[i]=lower_bound(q+1,q+len+1,w[i])-q;
    root[0]=build(1,len);
    for(int i=1;i<=n;i++) root[i]=update(w[i],1,len,root[i-1]);
    int x,y,z;
    while(m--){
        scanf("%d%d%d",&x,&y,&z);
        int id=query(root[x-1],root[y],1,len,z);
        printf("%d\n",q[id]);
    }
    return 0;
}

第二种直接插入。

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#include<queue>
using namespace std;
typedef long long ll;
const int N=2e5+10;
int ls[N*40],rs[N*40],val[N*40],sum[N*40];
int cnt=0,w[N],q[N],root[N*40];
void update(int &now,int l,int r,int x)
{
    ls[++cnt]=ls[now],rs[cnt]=rs[now];
    sum[cnt]=sum[now]+1;now=cnt;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(x<=mid) update(ls[now],l,mid,x);
    else update(rs[now],mid+1,r,x);
}
int query(int lrt,int rrt,int l,int r,int k)
{
    if(l==r) return l;
    int t=sum[ls[rrt]]-sum[ls[lrt]];
    int mid=(l+r)>>1;
    if(k<=t) return query(ls[lrt],ls[rrt],l,mid,k);
    else return query(rs[lrt],rs[rrt],mid+1,r,k-t);
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]),q[i]=w[i];
    sort(q+1,q+n+1);
    int len=unique(q+1,q+n+1)-q-1;
    for(int i=1;i<=n;i++) w[i]=lower_bound(q+1,q+len+1,w[i])-q;
    for(int i=1;i<=n;i++) root[i]=root[i-1],update(root[i],1,len,w[i]);
    int x,y,z;
    while(m--){
        scanf("%d%d%d",&x,&y,&z);
        int id=query(root[x-1],root[y],1,len,z);
        printf("%d\n",q[id]);
    }
    return 0;
}