poj 2104 区间第k大

最新推荐文章于 2021-09-22 17:20:36 发布

原创最新推荐文章于 2021-09-22 17:20:36 发布 · 301 阅读

0 ·

CC 4.0 BY-SA版权

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

1 篇文章

订阅专栏

本文详细解析了POJ 2104题目，探讨如何找到区间内的第k大元素。通过算法设计和实现，介绍了如何运用排序与比较策略来有效地解决这类问题，同时提供了完整的代码示例，帮助读者理解并掌握此类问题的解决技巧。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e5+10,logN=18;
struct Node
{
    int sum;
    Node *ch[2],*fa;
}node[N*logN],*root[N];
int tot=0,m,n,
a[N],b[N],size;
void build(Node* &t,Node *p,int l,int r,int delta)
{
    t=&node[++tot];
    *t=*p;
    t->sum++;
    if(l<r) {
        int m=(l+r)>>1;
        if(a[delta]<=b[m]) build(t->ch[0],p->ch[0],l,m,delta);
        else build(t->ch[1],p->ch[1],m+1,r,delta);
    }
}
int kth(Node *lc,Node *rc,int l,int r,int k)
{
    if(l==r) return b[l];
    int m=(l+r)>>1;
    int tmp=rc->ch[0]->sum-lc->ch[0]->sum;
    if(k<=tmp) return kth(lc->ch[0],rc->ch[0],l,m,k);
    else kth(lc->ch[1],rc->ch[1],m+1,r,k-tmp); 
}
int query(int l,int r,int k){return kth(root[l-1],root[r],1,size,k);}
int main()
{
    scanf("%d%d",&n,&m);
    fo(i,1,n) {
        scanf("%d",&a[i]);
        b[i]=a[i];
    }
    sort(b+1,b+1+n);
    size=unique(b+1,b+1+n)-b-1;
    root[0]=node;
    node->ch[0]=node->ch[1]=node;
    node->sum=0; 
    fo(i,1,n) build(root[i],root[i-1],1,size,i);
    while(m--) {
        int l,r,k;
        scanf("%d%d%d",&l,&r,&k);
        printf("%d\n",query(l,r,k));
    }
    return 0;
}