主席树是可持久化的线段树,其最大的优点就是在消耗内存较少的情况下保证了线段树的结构。
那么什么是主席树呢?对于区间[1,n],对于每一个子区间[1,i]建立一棵线段树(1<=i<=n),线段树的每一个节点存某个前缀[1..i]中属于区间[L..R]的数一共有多少个,这就是一颗主席树。
如果我们对于每一个区间都建一棵树,那肯定会爆内存,但由于区间[1,i]和区间[1,i+1]间仅有一个点的差别,所以线段树i和i+1也只有很小的差别,大部分点可以共用。只要在前一棵树上改动几个点,就可以形成新的树,这样就节省了许多的空间。
求区间K小,这个问题是一个经典的主席树问题,我们先讨论静态区间(动态区间需要树状数组套主席树才行)。
我们先把序列离散,排序。然后只需要建立n颗线段树就行,第i树维护[1,i]序列,这样我们处理任意区间l, r时就可以通过处理区间[1,l - 1], [1,r],就行,然后两者的处理结果进行相加相减就行。主席树的结果满足加减的性质,因为树l-1的点肯定也是树r的点。
静态区间K小代码:
#include<bits/stdc++.h>
#define MAXN 200005
using namespace std;
int read(){
char c;int x=0,y=1;
while(c=getchar(),(c<'0'||c>'9')&&c!='-');
if(c=='-') y=-1;else x=c-'0';
while(c=getchar(),c>='0'&&c<='9') x=x*10+c-'0';
return x*y;
}
struct edge{
int l,r,sum;
}T[MAXN*40];
struct node{
int num,pl;
}f[MAXN];
int n,m,cnt,rank[MAXN],root[MAXN];
int cmp(node a,node b){
return a.num<b.num?1:0;
}
void update(int p,int &x,int l,int r){
T[cnt++]=T[x];
x=cnt-1;
T[x].sum++;
if(l==r) return;
int mid=(l+r)>>1;
if(p<=mid) update(p,T[x].l,l,mid);
else update(p,T[x].r,mid+1,r);
}
int query(int i,int j,int k,int l,int r){
int fd=T[T[j].l].sum-T[T[i].l].sum;
if(l==r) return l;
int mid=(l+r)>>1;
if(k<=fd) return query(T[i].l,T[j].l,k,l,mid);
else return query(T[i].r,T[j].r,k-fd,mid+1,r);
}
int main()
{
n=read();m=read();
for(int i=1;i<=n;i++){
int x=read();
f[i]=(node){x,i};
}
sort(f+1,f+1+n,cmp);
cnt=1;T[0]=(edge){0,0,0};root[0]=0;
for(int i=1;i<=n;i++) rank[f[i].pl]=i;
for(int i=1;i<=n;i++){
root[i]=root[i-1];
update(rank[i],root[i],1,n);
}
for(int i=1;i<=m;i++){
int l=read(),r=read(),k=read();
printf("%d\n",f[query(root[l-1],root[r],k,1,n)].num);
}
return 0;
}