区间第k小值logn方法划分树模板HDOJ 4417 Super Mario

http://acm.hdu.edu.cn/showproblem.php?pid=4417

------ 划分树 + 二分 

划分树模板求区间第k小数，那么我们每次询问时就二分查找当前H算第几小数，那么显然答案就出来了~需要注意的就是区间全包括和全不包括需要特判一下，不然会RE。。。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <algorithm>
#include <string>
#include <cstring>
#define MID(x,y) ((x+y)>>1)

using namespace std;
#define M 100010
struct Seg_Tree
{
    int left,right;
    int mid()
    {
        return (left + right) >> 1;
    }
}tt[M<<2];
int len;
int sorted[M];
int toLeft[40][M];
int val[40][M];

void build(int l,int r,int d,int idx)
{
    tt[idx].left = l;
    tt[idx].right = r;
    if(tt[idx].left == tt[idx].right)    return ;
    int mid = tt[idx].mid();
    int lsame = mid - l + 1;//lsame表示和val_mid相等且分到左边的
    for(int i = l ; i <= r ; i ++)
    {
        if(val[d][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)
        {
            toLeft[d][i] = 0;//toLeft[i]表示[ tt[idx].left , i ]区域里有多少个数分到左边
        }
        else
        {
            toLeft[d][i] = toLeft[d][i-1];
        }
        if(val[d][i] < sorted[mid])
        {
            toLeft[d][i] ++;
            val[d+1][lpos++] = val[d][i];
        }
        else if(val[d][i] > sorted[mid])
        {
            val[d+1][rpos++] = val[d][i];
        }
        else
        {
            if(same < lsame)
            {//有lsame的数是分到左边的
                same ++;
                toLeft[d][i] ++;
                val[d+1][lpos++] = val[d][i];
            }
            else
            {
                val[d+1][rpos++] = val[d][i];
            }
        }
    }
    build(l,mid,d+1,idx<<1);
    build(mid+1,r,d+1,idx<<1|1);
}

int query(int l,int r,int k,int d,int idx) {
    if(l == r)
    {
        return val[d][l];
    }
    int s;//s表示[ l , r ]有多少个分到左边
    int ss;//ss表示 [tt[idx].left , l-1 ]有多少个分到左边
    if(l == tt[idx].left)
    {
        s = toLeft[d][r];
        ss = 0;
    }
    else
    {
        s = toLeft[d][r] - toLeft[d][l-1];
        ss = toLeft[d][l-1];
    }
    if(s >= k)
    {//有多于k个分到左边,显然去左儿子区间找第k个
        int newl = tt[idx].left + ss;
        int newr = tt[idx].left + ss + s - 1;//计算出新的映射区间
        return query(newl,newr,k,d+1,idx<<1);
    }
    else
    {
        int mid = tt[idx].mid();
        int bb = l - tt[idx].left - ss;//bb表示 [tt[idx].left , 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,d+1,idx<<1|1);
    }
}

int BS(int r, int h, int L, int R)
{
    int l = 0;
    while(l < r)
    {
        int mid = MID(l,r);
        if (query(L,R,mid,0,1) > h)
            r = mid;
        else    l = mid + 1;
    }
    return l;
}

int main()
{
    //freopen("test.in","r+",stdin);

    int t,caseo = 1;
    scanf("%d", &t);
    while(t--)
    {
        printf("Case %d:\n",caseo ++);
        int n,m;
        scanf("%d%d",&n, &m);
        for (int i = 1;i <= n; i++)
        {
            scanf("%d", &val[0][i]);
            sorted[i] = val[0][i];
        }
        sort(sorted+1,sorted+n+1);
        build(1,n,0,1);
        for (int i = 0; i < m; i++)
        {
            int L,R,H;
            scanf("%d%d%d",&L, &R, &H);
            if (query(L+1,R+1,1,0,1) > H )
                puts("0");
            else if (query(L+1,R+1,R-L+1,0,1) <= H)
                printf("%d\n",R-L+1);
            else
            {
                int res = BS(R-L+1,H,L+1,R+1);
                while(res != 0 && query(L+1,R+1,res,0,1) > H)
                    res--;
                printf("%d\n",res);

            }
        }
    }
    return 0;
}