划分树讲解

搜索关于划分树的讲解，竟然搜到了郑大媛学姐的博文，不是这篇

原博文有错误，我就不给出原链接了

用划分树来解决选定区间内的第K大值，其实也就两步！一步是建树，另一步则是查询。

     先说我对建树的理解吧！

    建树的过程很简单：两步就OK了！

   第一步：找到序列的中位数，把大于中位数的扔到中位数的左边，小于中位数的扔到数的右边。这样整个序列就被分成了两个区间。

   第二步：对每个子区间，也分别执行第一步操作，直到序列中只有一个元素为止。

   可以看出，建树是一个递归的过程，与线段树的建树有相似之处。

   划分树的建树需要注意以下几点：

     第一：建树是分层的，所以代码中用的是二维数组tree[20][M]。一般10W级别的数据，20层已经够了。

     第二：建树划分的标准是中位数，所以需要排序。而且只排一次序就OK了，为什么只排一次就OK了，我很久都没明白这一点。其实是这样的：对于任意序列： 划分后，左边的数据永远不会大于右边的数据。那么对左边数据单独排序与整体排序的结果是一样的，所以排一次序就OK了！

     第三：划分树划分好的数据永远在存放在下一层。比如数据：

tree[0][M]=1 5 2 6 3 7 4

排序后为：1 2 3 4 5 6 7

中位数为：4

划分后的结果为：tree[1][M]=1 2 3 4 5 6 7(这组数据有点特殊，划分后来就已经是排好序的了)红黑色字体都仍按原未排顺序排列

（红色表示划分到中位数的左边，黑色表示划分到中位数的右边）

接着划分：tree[2][M]=1 2 3 4 5 6 7

再接着分：tree[3][M]=1 2 3 4 5 6 0

到这里已经分完了，为什么最后是0呢？在第2层（tree[2][M]）,7已经分完了，所以不用再分

 第四：划分到最后，实际上已经对序列进行排序了。

     划分的时候还有一点需要处理：如果有多个数据相同怎么办呢？通过一种特殊的处理：尽量使左右两边平均分配相同的数。这个特殊处理是这样的：

    在没分之前，先假设中位数左边的数据suppose都已经分到左边了，所以suppose=mid-left+1;然后如果真的分在左边，即if(tree[level][i]<sorted[mid])

suppose--；suppose就减一！到最后，如果suppos=1，则说明中位数左边的数都小于中位数，如果有等于中位数的，则suppose大于1。

    最后分配的时候，把suppose个数，分到左边就可以了，剩下的分到右边！因为suppose的初值是mid-left+1,这样就能保证中位数左边和右边的数平衡了！

   第五：划分的过程，需要把每层的数据记录：toLeft[20][M]。toLeft[i][j]定义为：第i层[l,j]之间有多个数据被分到了左边（注意这里用的是闭区间）

模板：

[cpp] view plain copy

 print?

1. #include<stdio.h>    
2. #include<algorithm>    
3. using namespace std;    
4. #define M 100005    
5. int tree[20][M],sorted[M];    
6. int toLeft[20][M];    
7. void build(int level,int left,int right){    
8.     if(left==right)return ;    
9.     int mid=(left+right)>>1;    
10.     int i;        
11.     int suppose;//假设在中位数sorted[mid]左边的数都全部小于sorted[mid]    
12.     suppose=mid-left+1;    
13.     for(i=left;i<=right;i++){    
14.         if(tree[level][i]<sorted[mid]){    
15.             suppose--;    
16.         }    
17.     }    
18. /*如果suppose==1，则说明数组中值为sorted[mid]只有一个数。 
19.     比如序列：1 3 4 5 6，sorted[mid]=4   
20. */   
21. /*如果suppose>1，则说明数组中左半边值为sorted[mid]的不止一个数,为mid-suppose。 
22.     比如序列：1 4 4 4 6，sorted[mid]=4  
23. */    
24.     int lpos=left,rpos=mid+1;    
25.     for(i=left;i<=right;i++){    
26.         if(i==left){//这里是预处理，相当与初始化    
27.             toLeft[level][i]=0;    
28.         }  
29.         else{    
30.             toLeft[level][i]=toLeft[level][i-1];    
31.         }    
32.         if(tree[level][i]<sorted[mid]){//划分到中位数左边    
33.             toLeft[level][i]++;    
34.             tree[level+1][lpos++]=tree[level][i];    
35.         }  
36.         else if(tree[level][i]>sorted[mid]){//划分到中位数右边    
37.             tree[level+1][rpos++]=tree[level][i];    
38.         }  
39.         else{//这里，suppose大于0的数划分到中位数的左边    
40.             if(suppose!=0){//这里的处理太巧妙了！帅气！    
41.                 suppose--;    
42.                 toLeft[level][i]++;    
43.                 tree[level+1][lpos++]=tree[level][i];    
44.             }  
45.             else{//表示    
46.                 tree[level+1][rpos++]=tree[level][i];    
47.             }    
48.         }    
49.     }    
50.     build(level+1,left,mid);    
51.     build(level+1,mid+1,right);    
52. }    
53. //在[left,right]数据中查询[qleft,qright]中第k大的数据    
54. int query(int level,int left,int right,int qleft,int qright,int k){    
55.     if( left==right)    
56.         return tree[level][left];    
57.     int s;//代表[left,qleft)之间有多个个元素被分到左边    
58.     int ss;//[qleft, qright]内将被划分到左子树的元素数目    
59.     int mid=(left+right)>>1;    
60.     if(left==qleft){    
61.         s=0;    
62.         ss=toLeft[level][qright];    
63.     }else{    
64.         s=toLeft[level][qleft-1];    
65.         ss=toLeft[level][qright]-s;    
66.     }    
67.     int newl,newr;    
68.     if(k<=ss){//查询左边    
69.         newl=left+s;    
70.         newr=left+s+ss-1;    
71.         return query(level+1,left,mid,newl,newr,k);    
72.     }  
73.     else{//查询右边    
74.         newl=mid-left+1+qleft-s;    
75.         newr=mid-left+1+qright-s-ss;    
76.         return query(level+1,mid+1,right,newl, newr,k - ss);    
77.     }    
78. }    
79. int main(){    
80.     int n,m;    
81.     while(scanf("%d %d",&n,&m)!=EOF) {    
82.         int i;    
83.         for(i=1;i<=n;i++){    
84.             scanf("%d",&tree[0][i]);    
85.             sorted[i]=tree[0][i];    
86.         }    
87.         sort(sorted+1,sorted+n+1);    
88.         build(0,1,n);    
89.         int ql,qr,k;    
90.         for(i=0;i<m;i++){    
91.             scanf("%d %d %d",&ql,&qr,&k);    
92.             printf("%d\n",query(0,1,n,ql,qr,k));    
93.         }    
94.     }    
95.     return 0;    
96. }