区间第K值(可持久化线段树)

来源:互联网 发布:柠檬网络电视 | 柠檬tv 编辑:程序博客网 时间:2024/05/18 00:54

可持久化数据结构的一个经典应用
可以说是建k个线段树,但只用nlogn个节点
代码实现

#include<bits/stdc++.h>using namespace std;#define FOR(i,x,y) for(int i=(x),i##_end__=(y);i<=i##_end__;i++)#define DOR(i,x,y) for(int i=(x),i##_end__=(y);i>=i##_end__;i--)#define M 30005 struct Chairman_Tree{    static const int N=600005;    int tot;    int Lson[N],Rson[N];    int sum[N];    void init(){tot=0;}    void insert(int od,int l,int r,int &pre,int x){        pre=++tot;        Lson[pre]=Lson[od];        Rson[pre]=Rson[od];        sum[pre]=sum[od]+1;        if(l==r)return;        int mid=(l+r)>>1;        if(mid>=x)insert(Lson[od],l,mid,Lson[pre],x);        else insert(Rson[od],mid+1,r,Rson[pre],x);    }    int query(int l,int r,int lt,int rt,int x){        if(l==r||lt==rt)return l;        int res=sum[Lson[rt]]-sum[Lson[lt]];        int mid=(l+r)>>1;        if(res>=x)return query(l,mid,Lson[lt],Lson[rt],x);        else return query(mid+1,r,Rson[lt],Rson[rt],x-res);    }}K;int A[M],B[M];int hd[M];int main(){    int n,m;    cin>>n>>m;    FOR(i,1,n)scanf("%d",&A[i]),B[i]=A[i];    sort(B+1,B+1+n);        int len=unique(B+1,B+n+1)-B-1;    FOR(i,1,n){        int id=lower_bound(B+1,B+len+1,A[i])-B;        K.insert(hd[i-1],1,len,hd[i],id);    }    FOR(i,1,m){        int a,b,c;        scanf("%d%d%d",&a,&b,&c);        c=b-a-c+2;        printf("%d\n",B[K.query(1,len,hd[a-1],hd[b],c)]);    }    return 0;}
阅读全文
1 0