poj 2104 -- K-th Number (主席树模版题)

询问区间第k大问题，是主席树主要解决的一类问题。

下面说一下我对主席树这种数据结构的理解 ： 主席树就是可持久化线段树或者叫函数式线段树，这种数据结构支持访问历史版本。对于这个题而言我们最开始的想法是先对数据进行离散化然后建立n棵线段树，每一棵线段树 表示区间 1 到 i 之间 有哪些数(如果这个区间内有总排名第 i 打的数 那么 [i,i] 这个区间就设置成 1 否则设置成 0，这样的话查询区间和就可以得到排名在 ［l,r] 之间有多少个数) 但是对于这样的建树方法 会 MLE + TLE 。于是我们发现每次建新树的时候，只会有一个叶子节点的值从 0 变到1，加上其祖先节点最多一条链，由于线段树是一种类似于完全二叉树的数据结构，所以一条链就是logn个节点，这样我们就可以充分利用前面一个树的信息多开logn个节点去建立一棵新的线段树。我们不难发现，每一棵线段树的形态和大小都是一样的，所以他们具有相加减的特性 (并且我们可以发现他们相减就是 从区间 l 到 r 的所有数出现的位置和个数，这样我们就可以去询问第k打的数了)；

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <vector>
using namespace  std;
const int maxn = 1e5 + 10;
struct node {
    int l,r,sum;
}tr[maxn * 40];
vector <int> v;
int a[maxn] = {0};
int root[maxn] = {0};
int cnt = 0;
int getid (int x) {
    return lower_bound (v.begin(),v.end(),x) - v.begin() + 1;
}
void update (int l,int r,int x,int & y,int pos) {
    tr[++cnt] = tr[x],tr[cnt].sum ++,y = cnt;
    if (l == r) return ;
    int mid = (l + r) >> 1;
    if (pos <= mid) update(l, mid, tr[x].l, tr[y].l, pos);
    else update(mid + 1, r, tr[x].r, tr[y].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 = tr[tr[y].l].sum - tr[tr[x].l].sum;
    if (sum >= k) return query(l, mid, tr[x].l, tr[y].l, k);
    else return query(mid + 1, r, tr[x].r, tr[y].r, k - sum);
}
int main () {
    ios_base :: sync_with_stdio(false);
    int n,m;
    cin >> n >> m;
    for (int i = 1;i <= n; ++ i) {
        cin >> a[i];
        v.push_back(a[i]);
    }
    sort (v.begin(),v.end());
    for (int i = 1;i <= n; ++ i) update(1, n, root[i - 1], root[i], getid (a[i]));
    for (int i = 0;i < m ;++ i) {
        int l,r,k;
        cin >> l >> r >> k;
        cout << v[query(1, n, root[l-1], root[r], k) - 1] << endl;
    }
    return 0;
}