hdu4417（单纯裸划分树）

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

题意：给一些数，数中有重复的。还有一些询问，问的是[L,R] 区间内有多少个数小于h,有多次询问。

一般的方法肯定很容易超时，那这题就想到了划分树，接触过划分树的同学可能都知道用划分树查询区间内排列第k的数，这里只需把查询换为数值为k。用一个num表示小于等于k的数的个数，当k大于等于中位数就递归到右子数，同时num加上该区间内在左子数的个数

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=1e5+20;
int tree[20][N],sorted[N];
int toleft[20][N];
void build(int l,int r,int deq)
{
    if(l==r)
        return;
    int mid=(l+r)>>1;
    int same=mid-l+1;
    for(int i=l;i<=r;i++)
        if(tree[deq][i]<sorted[mid])
            same--;
    int L=l,R=mid+1;
    for(int i=l;i<=r;i++)
    {
        if(i==l)
            toleft[deq][i]=0;
        else
            toleft[deq][i]=toleft[deq][i-1];
        if(tree[deq][i]<sorted[mid]||(tree[deq][i]==sorted[mid]&&same>0))
        {
            tree[deq+1][L++]=tree[deq][i];
            toleft[deq][i]++;
            if(tree[deq][i]==sorted[mid])
                same--;
        }
        else
            tree[deq+1][R++]=tree[deq][i];
    }
    build(l,mid,deq+1);
    build(mid+1,r,deq+1);
}
int num;
void query(int L,int R,int l,int r,int k,int deq)
{
    if(l>r)
        return;
    if(l==r)
    {
        if(tree[deq][l]<=k)
            num++;
        return ;
    }
    int mid=(L+R)>>1;
    int to1,to2;
    if(l==L)
        to1=0;
    else
        to1=toleft[deq][l-1];
    to2=toleft[deq][r]-to1;
    int mm=(l+r)>>1;
    if(sorted[mid]<=k)//进入右子树查询
    {
        num+=to2;
        int lpos=l-1-L+1-to1,rpos=r-l+1-to2;//cout<<sorted[mid]<<" "<<num<<" "<<to2<<" "<<mid+1+lpos<<" "<<mid+1+lpos+rpos-1<<endl;
        query(mid+1,R,mid+1+lpos,mid+1+lpos+rpos-1,k,deq+1);
    }
    if(sorted[mid]>k)
        return query(L,mid,L+to1,L+to1+to2-1,k,deq+1);
}
int main()
{
    int T,q,l,r;
    scanf("%d",&T);
    int p=1;
    while(T--)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&tree[0][i]);
            sorted[i]=tree[0][i];
        }
        sort(sorted+1,sorted+1+n);
        build(1,n,0);
        int a,b,k;
        printf("Case %d:\n",p++);
        for(int j=1;j<=m;j++)
        {
            num=0;
            scanf("%d%d%d",&a,&b,&k);
            query(1,n,a+1,b+1,k,0);
            printf("%d\n",num);
        }
        //printf("\n");
    }
}