主席树(可持久化线段树)静态区间第K小

传送门主席树

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=2e5+5;
int n,m,a[maxn];
vector<int> v;//从0开始编号
inline int getid(int x)//得到值为x的编号
{
    return lower_bound(v.begin(),v.end(),x)-v.begin()+1;//lower_bound找到第一个值为x的位置,求下标要减去起始位置,由于从1开始计数,所以加1
}
typedef struct Node
{
    int l,r,sum;
}no;
no hjt[maxn*40];//*40记住就行,hjt以人名的字母开头,主席树hjt;
int cnt,root[maxn];
 
void insert(int l,int r,int pre,int &now,int p)//插入,最小值l,最大值r,,pre是上一棵线段树当前节点的编号,now是现在这个,p是新增的值
{
    hjt[++cnt]=hjt[pre];//++cnt 是新建当前节点的编号,这个式子是把上一棵线段树的节点信息传给当前节点
    now=cnt;
    hjt[now].sum++;//数目加加
    if(l==r) return;
    int m=(l+r)>>1;//右移,相当于除以2
    if(p<=m) insert(l,m,hjt[pre].l,hjt[now].l,p);
    else insert(m+1,r,hjt[pre].r,hjt[now].r,p);
}
 
int query(int l,int r,int L,int R,int k)//L,R分别表示要查询的两棵线段树的相应结点编号
{
    if(l==r) return l;//如果节点内只有一个值,这就是第K大,直接返回
    int m=(l+r)>>1;
    int t=hjt[hjt[R].l].sum-hjt[hjt[L].l].sum;//求当前插入的左子树的个数
    if(k<=t) return query(l,m,hjt[L].l,hjt[R].l,k);
    else return query(m+1,r,hjt[L].r,hjt[R].r,k-t);
}
 
signed main()
{
    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        v.push_back(a[i]);
    }
    sort(v.begin(),v.end());//从小到大开始排序+
    v.erase(std::unique(v.begin(),v.end()),v.end());//去重,
    for(int i=1;i<=n;i++)
    {
        insert(1,n,root[i-1],root[i],getid(a[i]));
    }
    while(m--)
    {
        int l,r,k;
        scanf("%lld%lld%lld",&l,&r,&k);
        int id=query(1,n,root[l-1],root[r],k)-1;//历史版本l-1,当前版本r;
        int ans=v[id];
        printf("%lld
",ans);
    }
    return 0;
}

原文地址:https://www.cnblogs.com/xcfxcf/p/12301655.html