POJ2104求區間第K大（主席樹）

#include <iostream>
#include <stdlib.h>
#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <map>
#include <vector>
#include <string.h>
#include <string>
#include <queue>
#include <set>
using namespace std;
//ios_base::sync_with_stdio(false);
//#include <bits/stdc++.h>
typedef long long ll;
const int maxn = 1e5+5;
int n, m, cnt, root[maxn], a[maxn], x, y, k;
struct Node {
    int l, r, sum;
}T[maxn*40];
vector <int> v;
int getid(int x)
{
    return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}
void update(int l ,int r, int &cur, int pre, int pos)
{
    T[++cnt] = T[pre],
    T[cnt].sum++,
    cur = cnt;
    if(l == r) return;
    int mid = (l+r)>>1;
    if(mid>=pos) update(l, mid, T[cur].l,T[pre].l, pos);
    else update(mid+1, r, T[cur].r, T[pre].r, pos);

}
int query(int l ,int r, int x, int y, int k)
{
    if(l == r) return l;
    int mid = (l+r)>>1;
    int sum = T[T[y].l].sum - T[T[x].l].sum;
    if(sum>=k) return query(l, mid, T[x].l, T[y].l, k);
    else return query(mid+1, r, T[x].r, T[y].r, k-sum);
}
int  main()
{
    scanf("%d%d", &n, &m);
    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]));
    }
    for(int i = 1; i <= m; i++)
    {
        scanf("%d%d%d", &x, &y, &k);
        printf("%d\n",v[query(1, n, root[x-1], root[y], k)-1]);

    }
}