poj2104函数式线段树

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
#define maxn 3000000
#define maxm 200001
struct node{
 int l,r,s,a,b;
}a[maxn];
struct P{
 int x,y;
 bool operator<(P a)const{
  return y<a.y;
 }
}b[maxm];
int n,m;
int tot;
int id[maxm],root[maxm];
int build(int l,int r){
 int k=++tot;
 a[k].a=l;
 a[k].b=r;
 a[k].s=0;
 if(l==r)return k;
 int mid=(l+r)>>1;
 a[k].l=build(l,mid);
 a[k].r=build(mid+1,r);
 return k;
}
int change(int p,int x,int w){
 int k=++tot;
 a[k].a=a[p].a,a[k].b=a[p].b,a[k].l=a[p].l,a[k].r=a[p].r;
 a[k].s=a[p].s+w;
 if(a[k].a==a[k].b)return k;
 int mid=(a[k].a+a[k].b)>>1;
 if(mid>=x){
  a[k].l=change(a[p].l,x,w);
 }else{
       a[k].r=change(a[p].r,x,w);
 }
 return k;
}
int res;
void query(int p,int q,int k){
 if(a[p].a==a[p].b){
 res=a[p].a;return; 
 }
 int pp=a[a[q].l].s-a[a[p].l].s;
 if(pp>=k){
  query(a[p].l,a[q].l,k);
 }else{
  query(a[p].r,a[q].r,k-pp);
 }
}
int main(){
 scanf("%d%d",&n,&m);
 for(int i=1;i<=n;i++)scanf("%d",&b[i].y),b[i].x=i;
 sort(b+1,b+n+1);
 for(int i=1;i<=n;i++)id[b[i].x]=i;
 root[0]=build(1,n);
 for(int i=1;i<=n;i++){
  root[i]=change(root[i-1],id[i],1);
 }
    for(int i=1;i<=m;i++){
  int x,y,z;
  scanf("%d%d%d",&x,&y,&z);
  query(root[x-1],root[y],z);
  printf("%d\n",b[res].y);
    }
}