C++ 代码
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 100010,M = 10010;
int n,m;
int a[N];
vector<int> nums;
struct node
{
int l,r;
int cnt;
}tr[N*4+N*17];
int root[N],idx;
int find(int x)
{
return lower_bound(nums.begin(),nums.end(),x)-nums.begin();
}
int build(int l,int r)
{
int p=++idx;
if(l==r)return p;
int mid=l+r>>1;
tr[p].l=build(l,mid),tr[p].r=build(mid+1,r);
return p;
}
int insert(int p,int l,int r,int x)
{
int q=++idx;//q是新的版本节点。
tr[q]=tr[p];//似乎是必然开一个新节点,与可持久化trie树有些不同。这一条新的链必然更新成新的东西。
if(l==r)
{
tr[q].cnt++;
return q;
}
int mid=l+r>>1;
if(x<=mid)tr[q].l=insert(tr[p].l,l,mid,x);
else tr[q].r=insert(tr[p].r,mid+1,r,x);
tr[q].cnt=tr[tr[q].l].cnt+tr[tr[q].r].cnt;
return q;
}
int query(int q,int p,int l,int r, int k)//p为新版本,q为老版本
{
if(l==r)return r;
int cnt=tr[tr[q].l].cnt-tr[tr[p].l].cnt;
int mid=l+r>>1;
if(k<=cnt)return query(tr[q].l,tr[p].l,l,mid,k);//向左边递归(因为查出的数量已经是足够的啦)
else return query(tr[q].r,tr[p].r,mid+1,r,k-cnt);//向右边递归(因为查出的元素还不够,把左边减下去在继续从右边找);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
nums.push_back(a[i]);
}
sort(nums.begin(),nums.end());
nums.erase(unique(nums.begin(),nums.end()),nums.end());
root[0]=build(0,nums.size()-1);
for(int i=1;i<=n;i++)
{
root[i]=insert(root[i-1],0,nums.size()-1,find(a[i]));//i-1相当于是老版本。
}
while(m--)
{
int l,r,k;
scanf("%d%d%d",&l,&r,&k);
printf("%d\n",nums[query(root[r],root[l-1],0,nums.size()-1,k)]);//查询0到nums. 里面这个区间的两种版本差值,里面还会有二分,直接解里面已经求好的和做一个二分就是能快速查询,
}
return 0;
}