POJ2104（主席树求区间第K大）

模板题，模板来自B站UESTCACM

#include <iostream>
#include <algorithm>
#include <queue>
#include <stack>
#include <cstdio>
#include <string>
#include <cstring>
#include <vector>
#include <set>
#include <map>
#include <sstream>
#include <cmath>
#include <stack>
#define LL long long
#define mod 1000000007
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e6 + 5;
vector<int> v;
int a[maxn], root[maxn];
int cnt;
int getid(int x){
    return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}
struct node{
    int l,r,sum;
}T[maxn*40];
void update(int l, int r, int &x, int y, int pos){
    T[++cnt] = T[y];
    T[cnt].sum++;
    x = cnt;
    if(l == r) return;
    int m = (l+r)/2;
    if(m >= pos) update(l,m,T[x].l,T[y].l,pos);
    else update(m+1,r,T[x].r,T[y].r,pos);
}
int query(int l, int r, int x, int y, int k){
    if(l == r) return l;
    int m = (l + r)/2;
    int sum = T[T[y].l].sum - T[T[x].l].sum;
    if(sum >= k) return query(l,m,T[x].l,T[y].l,k);
    else return query(m+1,r,T[x].r,T[y].r,k-sum);
}
int main(){
    int n,q;
    scanf("%d%d",&n,&q);
    for(int i=1; i<=n; i++){
        scanf("%d",&a[i]);
        v.push_back(a[i]);
    }
    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());
    for(int i=1; i<=n; i++)
        update(1,n,root[i],root[i-1],getid(a[i]));
    while(q--){
        int x,y,k;
        scanf("%d%d%d",&x,&y,&k);
        printf("%d\n",v[query(1,n,root[x-1],root[y],k)-1]);
    }
    return 0;

}