[POJ2104] K-th Number（归并树,二分）

题目链接：http://poj.org/problem?id=2104

题意：略

最近想学一下分块，搜到了这道题。结果在《挑战》上另外看到了另外的用线段树做的算法。后来才知道，这种记录归并排序过程的树叫归并树。利用STL，非常好实现。

思路就是首先二分答案，拿着这个答案在归并树上找对应区间比这个数小的数的个数ret，找到==k的为止。

稍微对拍后，重新划了一下query的区间才发现是第40行写搓了。最近不刷题，码力下降了啊。。

#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <climits>
#include <complex>
#include <cassert>
#include <cstdio>
#include <bitset>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <ctime>
#include <set>
#include <map>
#include <cmath>
using namespace std;

#define lrt rt << 1
#define rrt rt << 1 | 1
const int maxn = 100100;
int n, q, a[maxn];
vector<int> seg[maxn<<2];

void build(int l, int r, int rt) {
    if(l == r) {
        seg[rt].clear();
        seg[rt].push_back(a[l]);
        return;
    }
    int mid = (l + r) >> 1;
    build(l, mid, lrt);
    build(mid+1, r, rrt);
    seg[rt].resize(r-l+1);
    merge(seg[lrt].begin(), seg[lrt].end(), seg[rrt].begin(), seg[rrt].end(), seg[rt].begin());
}

int query(int L, int R, int v, int l, int r, int rt) {
    if(R < l || r < L) return 0;
    if(L <= l && r <= R) return upper_bound(seg[rt].begin(), seg[rt].end(), v) - seg[rt].begin();
    int mid = (l + r) >> 1;
    return query(L, R, v, l, mid, lrt) + query(L, R, v, mid+1, r, rrt);
}

inline bool scan_d(int &num) {
    char in;bool IsN=false;
    in=getchar();
    if(in==EOF) return false;
    while(in!='-'&&(in<'0'||in>'9')) in=getchar();
    if(in=='-'){ IsN=true;num=0;}
    else num=in-'0';
    while(in=getchar(),in>='0'&&in<='9'){
        num*=10,num+=in-'0';
    }
    if(IsN) num=-num;
    return true;
}

inline void out(int x) { 
    if (x > 9) out(x / 10); 
    putchar(x % 10 + '0');
}

int main() {
    // freopen("in", "r", stdin);
    // freopen("out", "w", stdout);
    int T, l, r, k;
    while(scan_d(n)) {
        scan_d(q);
        for(int i = 1; i <= n; i++) scan_d(a[i]);
        build(1, n, 1);
        sort(a+1, a+n+1);
        while(q--) {
            scan_d(l); scan_d(r); scan_d(k);
            int lo = 1, hi = n;
            int v;
            while(lo <= hi) {
                int mid = (lo + hi) >> 1;
                int ret = query(l, r, a[mid], 1, n, 1);
                if(ret >= k) {
                    hi = mid - 1;
                    v = a[mid];
                }
                else lo = mid + 1;
            }
            printf("%d
", v);
        }
    }
    return 0;
}