ACM——树

1、主席树

///静态区间K小值查询
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int a[N],b[N];
int q,sz,p,n,m;
int lc[N<<5],rc[N<<5],sum[N<<5],rt[N<<5];
/*a、b数组，一般储存输入数据
sz：节点个数
rt数组：存储每棵线段树的根节点编号
lc、rc数组：记录左儿子、右儿子编号，类似于动态开点
sum数组：记录节点权值
p：记录离散化后序列长度，也是线段树的区间最大长度*/
void build(int &rt,int l,int r)///&rt 引用
{
    rt=++sz,sum[rt]=0;
    if(l==r) return;
    int mid=(l+r)/2;
    build(lc[rt],l,mid);///rt右子树
    build(rc[rt],mid+1,r);///rt左子树
}
int update(int o,int l,int r)///返回新节点编号
{
    int oo=++sz;
    lc[oo]=lc[o];///继承原来的左右儿子
    rc[oo]=rc[o];
    sum[oo]=sum[o]+1;///更新结点值
    if(l==r) return oo;
    int mid=(l+r)/2;
    if(mid>=p)
        lc[oo]=update(lc[oo],l,mid);
    else
        rc[oo]=update(rc[oo],mid+1,r);
    return oo;
}
int query(int u,int v,int l,int r,int k)
{
    int mid=(l+r)/2;
    int x=sum[lc[v]]-sum[lc[u]];///区间[u,v]形成新的树里面左子树有多少数
    if(l==r) return l;
    if(x>=k) return query(lc[u],lc[v],l,mid,k);///在左边找
    else return query(rc[u],rc[v],mid+1,r,k-x);
}
int main()
{
    ios_base::sync_with_stdio(NULL);
    cin.tie(NULL),cout.tie(NULL);
    int l,r,k;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>a[i],b[i]=a[i];
    sort(b+1,b+n+1);
    q=unique(b+1,b+n+1)-b-1;///有多少不重复数据
    build(rt[0],1,q);///建空树
    for(int i=1;i<=n;i++)///加入数字
    {
        p=lower_bound(b+1,b+1+q,a[i])-b;///他前面有多少小于他的数
        rt[i]=update(rt[i-1],1,q);
    }
    while(m--)
    {
        cin>>l>>r>>k;
        cout<<b[query(rt[l - 1], rt[r], 1, q, k)]<<endl;
    }
    return 0;
}