主席树求静态区间第K大

#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 100005
using namespace std;
int N,M;
struct LS
{
    int x,id;
}a[MAXN];
struct Node
{
    int l,r,cnt;
}Seg[MAXN*20];
int mp[MAXN],f[MAXN];
inline bool cmp(LS A,LS B)
{
    return A.x<B.x;
}
int root[MAXN],tot=0,cnt=1;
inline int build(int l,int r)
{
    int k=++tot;
    if(l==r)
    {
        Seg[k].cnt=0;
        return k;
    }
    int mid=(l+r)>>1;
    Seg[k].l=build(l,mid);
    Seg[k].r=build(mid+1,r);
    return k;
}
inline int insert(int pre,int l,int r,int pos)
{
    int k=++tot;
    Seg[k]=Seg[pre];
    if(l==pos&&r==pos)
    {
        Seg[k].cnt+=1;
        return k;
    }
    int mid=(l+r)>>1;
    if(pos<=mid) Seg[k].l=insert(Seg[pre].l,l,mid,pos);
    else Seg[k].r=insert(Seg[pre].r,mid+1,r,pos);
    Seg[k].cnt=Seg[Seg[k].l].cnt+Seg[Seg[k].r].cnt;
    return k;
}
inline int query(int l,int r,int pre,int cur,int K)//区间内查询第K个数
{
    if(l==r) return l;
    int mid=(l+r)>>1;
    int res=Seg[Seg[cur].l].cnt-Seg[Seg[pre].l].cnt;
    if(K<=res)  return query(l,mid,Seg[pre].l,Seg[cur].l,K);
    else return query(mid+1,r,Seg[pre].r,Seg[cur].r,K-res);
}
inline void init()
{
    tot=0;cnt=1;
    memset(a,0,sizeof(a));
    memset(f,0,sizeof(f));
    memset(mp,0,sizeof(mp));
    memset(root,0,sizeof(root));
    memset(Seg,0,sizeof(Seg));
}
int main()
{
    while(scanf("%d%d",&N,&M)!=EOF)
    {
        //init();
        for(int i=1;i<=N;i++)
        {
            scanf("%d",&a[i].x);
            a[i].id=i;
        }
        sort(a+1,a+N+1,cmp);
        f[a[1].id]=cnt;mp[cnt]=a[1].x;
        for(int i=2;i<=N;i++)
        {
            if(a[i].x>a[i-1].x)
                cnt++;
            f[a[i].id]=cnt;
            mp[cnt]=a[i].x;
        }
        root[0]=build(1,cnt);
        for(int i=1;i<=N;i++)
            root[i]=insert(root[i-1],1,cnt,f[i]);
        int L,R,KK;
        for(int i=1;i<=M;i++)
        {
            scanf("%d%d%d",&L,&R,&KK);
            printf("%d\n",mp[query(1,cnt,root[L-1],root[R],KK)]);
        }
    }
    return 0;
}

阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/UFBoon/article/details/51560140
文章标签: c语言 算法
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭