poj 2104 K-th Number 划分树

poj 2104 K-th Number

//poj 2104 K-th Number
//划分树

//建树和查找，模板题，具体看代码

#define infile freopen("in.txt", "r", stdin);
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

#define N 100005

int ans;
int level[20][N], sorted[N], l_cnt[20][N];

void build(int l, int r, int loc, int dep)
{
    if(l == r)
        return;
    int mid = l + (r-l)/2;
    int mid_val = sorted[mid];

    int cnt = 0;    //记录小于mid_val的个数
    for(int i = l; i <= r; ++i)
        if(level[dep][i] < mid_val)
            cnt++;
    //这里l_tot是从l开始···equal_cnt记录跟mid_val相等且进入左子树的个数
    int l_tot = l, r_tot = mid+1, equal_cnt = 0;
    for(int i = l; i <= r; ++i)
    {
        if(level[dep][i] < mid_val) //小于mid_val的进入左子树
            level[dep+1][l_tot++] = level[dep][i];
        else if(level[dep][i] == mid_val && mid-l+1-cnt > equal_cnt)
        {   //等于mid_val的且没超过左子树限定个数的时候进入左子树
            equal_cnt++;
            level[dep+1][l_tot++] = level[dep][i];
        }
        else
            level[dep+1][r_tot++] = level[dep][i];
        l_cnt[dep][i] = l_tot - l;//记录进入左子树的个数
    }
    build(l, mid, loc*2, dep+1);
    build(mid+1, r, loc*2+1, dep+1);
}

void find(int l, int r, int from, int to, int k, int dep)
{
    if(l == r)
    {
        ans = level[dep][l];
        return;
    }
    int mid = l + (r-l)/2;
    int from_cnt;   //记录当前层中，from以前的数进入左子树的个数
    if(from > 1)    //不包括from
        from_cnt = l_cnt[dep][l+from - 2];
    else
        from_cnt = 0;
    //记录当前层中，to以前的数进入又子树的个数，包括to
    int to_cnt = l_cnt[dep][l+to - 1];


    //接下去那层就应该保持和这层一样的起点，即from左边的个数不变
    //若下一层到左子树，则要从from以左的进入左子树的个数(from_cnt)开始
    //由于from这个数也到左子树所以下一层应从from_cnt+1开始
    //若下一层到右子树要去掉进入左子树的个数
    if(to_cnt - from_cnt >= k)  //from到to之间进入左子树的个数
        find(l, mid, from_cnt+1, to_cnt, k, dep+1);
    else
        find(mid+1, r, from-from_cnt, to-to_cnt, k-(to_cnt-from_cnt), dep+1);
}

int main(void)
{
    //infile
    int n, n_query;
    while(scanf("%d%d", &n, &n_query) != EOF)
    {
        for(int i = 1; i <= n; ++i)
        {
            scanf("%d", &level[1][i]);
            sorted[i] = level[1][i];
        }
        sort(sorted+1, sorted+n+1);
        build(1, n, 1, 1);
        while(n_query--)
        {
            int from, to, k;
            scanf("%d%d%d", &from, &to, &k);
            find(1, n, from, to, k, 1);
            printf("%d\n", ans);
        }
    }
    return 0;
}

还不懂的话可以看看这个

http://www.cnblogs.com/pony1993/archive/2012/07/17/2594544.html

转载于:https://www.cnblogs.com/gabo/archive/2012/08/12/2634343.html