poj2761Feed the dogs(划分树-区间K值）

链接

这树着实不好理解啊

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

对于找K值 右区间的确定不是太理解。。先当模板贴着吧

 #include <iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<stdlib.h>
 using namespace std;
 #define N 100010
 int tree[][N],sum[][N];//每层的数  及每层截止到i会放进左子树的个数
 int cu[N],xu[N];//原数据 及排序后的数据
 void build(int c,int l,int r)
 {
     int i,m = (l+r)>>;
     int o = m-l+,rp = m+,lp = l;
     for(i = l ; i <= m ; i++)//左子树中与xu[m]相同的个数
     if(xu[i]<xu[m])
     o--;
     for(i = l ; i <= r ; i++)
     {
         if(i==l)
         sum[c][i] = ;
         else
         sum[c][i] = sum[c][i-];//维护一下 到I为止被分配到左子树中的个数
         if(tree[c][i]==xu[m])
         {
             if(o)
             {
                 sum[c][i]++;
                 o--;
                 tree[c+][lp++] = xu[m];//分配到左子树中
             }
             else
             tree[c+][rp++] = tree[c][i];
         }
         else if(tree[c][i]<xu[m])
         {
             sum[c][i]++;
             tree[c+][lp++] = tree[c][i];
         }
         else
         tree[c+][rp++] = tree[c][i];//分配到右子树中
     }
     if(l==r) return ;
     build(c+,l,m);
     build(c+,m+,r);
 }
 int query(int a,int b,int k,int c,int l,int r)
 {
     if(l==r)
     {
         return tree[c][l];
     }
     int m = (l+r)>>;
     int s,ss;
     if(a==l)//这里特殊处理一下 sum没初始化 不然不处理应该也可以
     {
         s = ;
         ss = sum[c][b];
     }
     else
     {
         s = sum[c][a-];
         ss = sum[c][b]-s;
     }
     if(k<=ss)
     {
         return query(l+s,l+s+ss-,k,c+,l,m);
     }
     else
     return query(m-l++a-s,m-l++b-s-ss,k-ss,c+,m+,r);

 }
 int main()
 {
     int i,n,m;
     while(scanf("%d%d",&n,&m)!=EOF)
     {
         for(i =  ; i <= n ; i++)
         {
             scanf("%d",&cu[i]);
             tree[][i] = cu[i];
             xu[i] = cu[i];
         }
         sort(xu+,xu+n+);
         build(,,n);
         while(m--)
         {
             int a,b,c;
             scanf("%d%d%d",&a,&b,&c);
             int ans = query(a,b,c,,,n);
             printf("%d\n",ans);
         }
     }
     return ;
 }