可持久化线段树--第k小数

洛谷传送门

思路：

对于给定序列的每个数，我们都用一颗线段树来维护添加了这个数的新版本。

然后利用前缀和的思想找到第k小数所在的叶子结点（因为我们创建新线段树的时候保证了新插入的数的顺序，所以只用判断每个节点的子树大小就能找到对应的第几个数）。

数据过大的时候注意要离散化，此题用容器map即可。因为每次开一颗线段树很费空间，所以要进行空间优化：即动态开点，只开修改了数据的那条链的点。

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
int root[N];
struct xds
{
    int l,r,sum;
}xd[23*N];
int n,m;
map<int,int>k;
map<int,int>k_f;
vector<int>f;
int p;
void build(int &rt,int l,int r)
{
    if(!rt) rt=++p;
    if(l==r) return;
    int mid=l+r>>1;
    build(xd[rt].l,l,mid);
    build(xd[rt].r,mid+1,r);
}
void motify(int pre,int &rt,int l,int r,int pos)
{
    if(!rt) rt=++p;
    xd[rt].sum=xd[pre].sum+1;
    if(l==r) return;
    int mid=l+r>>1;
    if(pos<=mid)
    {
        xd[rt].r=xd[pre].r;
        motify(xd[pre].l,xd[rt].l,l,mid,pos);
    }
    else{
        xd[rt].l=xd[pre].l;
        motify(xd[pre].r,xd[rt].r,mid+1,r,pos);
    }
}
int query(int pre,int cur,int l,int r,int fsum)
{
    if(l==r) return l;
    int mid=l+r>>1;
    int zsum=xd[xd[cur].l].sum-xd[xd[pre].l].sum;
    if(fsum<=zsum) return query(xd[pre].l,xd[cur].l,l,mid,fsum);
    else return query(xd[pre].r,xd[cur].r,mid+1,r,fsum-zsum);
}
void solve(){
    cin>>n>>m;
    set<int>s;
    int val_2=0;
    for(int i=0;i<n;i++)
    {
        int x;
        cin>>x;
        f.push_back(x);
        s.insert(x);
    }
    for(auto v:s)
    {
        k[v]=++val_2;
        k_f[val_2]=v;
    }
    build(root[0],1,n);
  //  cout<<root[0]<<'\n';
    for(int i=0;i<n;i++)
    {
        motify(root[i],root[i+1],1,n,k[f[i]]);
    }
  //  for(int i=1;i<=n;i++) cout<<xd[root[i]].sum<<"\n";
    for(int i=1;i<=m;i++)
    {
        int l,r,k_2;
        cin>>l>>r>>k_2;
     //   cout<<l<<' '<<r<<' '<<k_2<<'\n';
        cout<<k_f[query(root[l-1],root[r],1,n,k_2)]<<"\n";
    }
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr),cout.tie(nullptr);
    int T=1;
    while(T--){solve();}
    return 0;
}