POJ2104 K-th Number 不带修改的主席树 线段树

时间:2024-04-29 00:12:55

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

给定一个序列,求区间第k小

通过构建可持久化的点,得到线段树左儿子和右儿子的前缀和(前缀是这个序列从左到右意义上的),然后是一个二分的get操作。

 #include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define LL long long
const LL maxn=;
using namespace std;
int n,m;
struct node{
int lc,rc,sum;
}t[maxn*];int tot;
struct nod{
int x,id;
}a[maxn];
int rk[maxn]={},rt[maxn]={};
bool mcmp(nod aa,nod bb){return aa.x<bb.x;}
void insert(int &x,int l,int r,int num){
t[++tot]=t[x]; ++t[tot].sum;x=tot;//cout<<t[x].sum<<endl;
if(l==r)return;
int mid=(r+l)/;
if(num<=mid)insert(t[x].lc,l,mid,num);
else insert(t[x].rc,mid+,r,num);
}
int getit(int x,int y,int k,int l,int r){
if(l==r)return l;
int z=t[t[y].lc].sum-t[t[x].lc].sum;
int mid=(r+l)/;
if(k<=z)return getit(t[x].lc,t[y].lc,k,l,mid);
else return getit(t[x].rc,t[y].rc,k-z,mid+,r);
}
int main(){
while(~scanf("%d%d",&n,&m)){
tot=;
for(int i=;i<=n;i++){scanf("%d",&a[i].x);a[i].id=i;}
sort(a+,a++n,mcmp);
for(int i=;i<=n;i++) rk[a[i].id]=i;
for(int i=;i<=n;i++){rt[i]=rt[i-]; insert(rt[i],,n,rk[i]);}
int x,y,k;
for(int i=;i<=m;i++){
scanf("%d%d%d",&x,&y,&k);
printf("%d\n",a[getit(rt[x-],rt[y],k,,n)].x);
}
}
return ;
}