关闭

主席树求静态区间第K大

标签: c语言算法
151人阅读 评论(0) 收藏 举报
#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;
}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:4362次
    • 积分:323
    • 等级:
    • 排名:千里之外
    • 原创:29篇
    • 转载:0篇
    • 译文:0篇
    • 评论:0条
    文章分类