POJ - 2104——K-th Number （主席树静态区间第k大）

最新推荐文章于 2019-09-17 14:06:42 发布

creator平

最新推荐文章于 2019-09-17 14:06:42 发布

阅读量117

点赞数

分类专栏：可持久化数据结构

本文链接：https://blog.csdn.net/weixin_41380961/article/details/100692631

版权

可持久化数据结构专栏收录该内容

8 篇文章 1 订阅

订阅专栏

区间第k大
给出n个数，m个查询，数列ai，查询l到r第k大的值
7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3

5
6
3

#include<string.h>
#include<stdio.h>
#include<algorithm>
#define id(p) p.second
#define M 100010
#define x(p) p.first
using namespace std;
struct segment_tire
{
    int l,r,sum;//l代表左子树，r为右子树，sum为节点数量
    segment_tire(int _l=0,int _r=0,int _sum=0):l(_l),r(_r),sum(_sum){};
}T[20*M];
pair<int,int> A[M];
int root[M],disc[M];//root为每个版本的线段树的根
int n,m,cut;
void update(int &rt,int x,int l,int r)
{
    T[++cut]=T[rt];
    rt=cut;
    T[rt].sum++;
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(x<=mid) update(T[rt].l,x,l,mid);
    else update(T[rt].r,x,mid+1,r);
}
int query(int i,int j,int k,int l,int r)
{
    int d=T[T[j].l].sum-T[T[i].l].sum;
    if(l==r) return l;
    int mid=(l+r)>>1;
    if(k<=d) return query(T[i].l,T[j].l,k,l,mid);
    else return query(T[i].r,T[j].r,k-d,mid+1,r);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x(A[i]));
        id(A[i])=i;
    }
    sort(A+1,A+1+n);
    for(int i=1;i<=n;i++)
        disc[id(A[i])]=i;//离散化后的数组,这里的离散化不去重
    root[0]=0,cut=0;
    T[0]=segment_tire(0,0,0);
    for(int i=1;i<=n;i++)
    {
        root[i]=root[i-1];
        update(root[i],disc[i],1,n);
    }
    int l,r,k;
    while(m--)
    {
        scanf("%d%d%d",&l,&r,&k);
        int ans=query(root[l-1],root[r],k,1,n);
        printf("%d\n",x(A[ans]));
    }
}