POJ 2104 & HDU 2665 K-th Number （劃分樹）

題意不用說了，就是詢問x次，指定區間的第k小的值....................

表示太弱了.......看了許久還是不太理解劃分樹內涵，先把偷來的模板留着

#include <iostream>
#include <algorithm>
#include <cmath>
#include<functional>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <climits>//形如INT_MAX一類的
#define MAX 100005
#define INF 0x7FFFFFFF
#define L(x) x << 1
#define R(x) x << 1 | 1
using namespace std;


struct Seg_Tree {
    int l,r,mid;
} tr[MAX*4];
int len;
int sorted[MAX];
int lef[20][MAX];
int val[20][MAX];

void build(int l,int r,int step,int x) {
    tr[x].l = l;
    tr[x].r = r;
    tr[x].mid = (l + r) >> 1;
    if(tr[x].l == tr[x].r) return ;
    int mid = tr[x].mid;
    int lsame = mid - l + 1;//lsame表示和val_mid相等且分到左邊的
    for(int i = l ; i <= r ; i ++) {
        if(val[step][i] < sorted[mid]) {
            lsame --;//先假設左邊的數(mid - l + 1)個都等於val_mid,然後把實際上小於val_mid的減去
        }
    }
    int lpos = l;
    int rpos = mid + 1;
    int same = 0;
    for(int i = l ; i <= r ; i ++) {
        if(i == l) {
            lef[step][i] = 0;//lef[i]表示[ tr[x].l , i ]區域裏有多少個數分到左邊
        } else {
            lef[step][i] = lef[step][i-1];
        }
        if(val[step][i] < sorted[mid]) {
            lef[step][i] ++;
            val[step + 1][lpos++] = val[step][i];
        } else if(val[step][i] > sorted[mid]) {
            val[step+1][rpos++] = val[step][i];
        } else {
            if(same < lsame) {//有lsame的數是分到左邊的
                same ++;
                lef[step][i] ++;
                val[step+1][lpos++] = val[step][i];
            } else {
                val[step+1][rpos++] = val[step][i];
            }
        }
    }
    build(l,mid,step+1,L(x));
    build(mid+1,r,step+1,R(x));
}

int query(int l,int r,int k,int step,int x) {
    if(l == r) {
        return val[step][l];
    }
    int s;//s表示[l , r]有多少個分到左邊
    int ss;//ss表示 [tr[x].l , l-1 ]有多少個分到左邊
    if(l == tr[x].l) {
        s = lef[step][r];
        ss = 0;
    } else {
        s = lef[step][r] - lef[step][l-1];
        ss = lef[step][l-1];
    }
    if(s >= k) {//有多於k個分到左邊,顯然去左兒子區間找第k個
        int newl = tr[x].l + ss;
        int newr = tr[x].l + ss + s - 1;//計算出新的映射區間
        return query(newl,newr,k,step+1,L(x));
    } else {
        int mid = tr[x].mid;
        int bb = l - tr[x].l - ss;//bb表示 [tr[x].l , l-1 ]有多少個分到右邊
        int b = r - l + 1 - s;//b表示 [l , r]有多少個分到右邊
        int newl = mid + bb + 1;
        int newr = mid + bb + b;
        return query(newl,newr,k-s,step+1,R(x));
    }
}

int n,m,l,r,k;
int main() {
    while(scanf("%d%d",&n,&m) != EOF) {
        for(int i=1; i<=n; i++) {
            scanf("%d",&val[0][i]);
            sorted[i] = val[0][i];
        }
        sort(sorted + 1,sorted + 1 + n);
        build(1,n,0,1);
        for(int i=0; i<m; i++) {
            scanf("%d%d%d",&l,&r,&k);
            printf("%d\n",query(l,r,k,0,1));
        }
    }
    return 0;
}