划分树
划分树是一种来解决区间第
建议先学完主席树再看划分树哦
建树¶
划分树的建树比较简单,但是相对于其他树来说比较复杂。
如图,每一层都有一个看似无序的数组。其实,每一个被红色标记的数字都是 要分配到左儿子的 。而分配的规则是什么?就是与 这一层的中位数 做比较,
我们不肯能每一次都对每一层排序,这样子不说常数,就算是理论复杂度也过不去。我们想,找中位数,一次排序就够了。为什么?比如,我们求
两个关键数组:
1 2 | tree[log(N),N] : 也就是树,要存下所有的值,空间复杂度 $O(n\log n)$。 toleft[log(N),n] : 也就是每一层 1~i 进入左儿子的数量,这里需要理解一下,这是一个前缀和。 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | procedure Build(left,right,deep:longint); // left,right 是左右区间,deep是第几层 var i,mid,same,ls,rs,flag:longint; // 其中 flag 是用来平衡左右两边的数量的 begin if left=right then exit; // 到底层了 mid:=(left+right) >> 1; same:=mid-left+1; for i:=left to right do if tree[deep,i]<num[mid] then dec(same); ls:=left; // 分配到左儿子的第一个指针 rs:=mid+1; // 分配到右儿子的第一个指针 for i:=left to right do begin flag:=0; if (tree[deep,i]<num[mid])or((tree[deep,i]=num[mid])and(same>0)) then // 分配到左边的条件 begin flag:=1; tree[deep+1,ls]:=tree[deep,i]; inc(ls); if tree[deep,i]=num[mid] then // 平衡左右个数 dec(same); end else begin tree[deep+1,rs]:=tree[deep,i]; inc(rs); end; toleft[deep,i]:=toleft[deep,i-1]+flag; end; Build(left,mid,deep+1); // 继续 Build(mid+1,right,deep+1); end; |
查询¶
那我们先扯一下主席树的内容。在用主席树求区间第
查询难理解的,在于 区间缩小 这种东西。下图,我查询的是
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | function Query(left,right,k,l,r,deep:longint):longint; var mid,x,y,cnt,rx,ry:longint; begin if left=right then // 写成 l=r 也无妨,因为目标区间也一定有答案 exit(tree[deep,left]); mid:=(l+r) >> 1; x:=toleft[deep,left-1]-toleft[deep,l-1]; // l 到 left 的去左儿子的个数 y:=toleft[deep,right]-toleft[deep,l-1]; // l 到 right 的去左儿子的个数 ry:=right-l-y; rx:=left-l-x; // ry 是 l 到 right 去右儿子的个数,rx 则是 l 到 lefr 去右儿子的个数 cnt:=y-x; // left 到 right 左儿子的个数 if cnt>=k then // 主席树常识啦 Query:=Query(l+x,l+y-1,k,l,mid,deep+1) // l+x 就是缩小左边界,l+y-1 就是缩小右区间。对于上图来说,就是把 1 和 2 放弃了。 else Query:=Query(mid+rx+1,mid+ry+1,k-cnt,mid+1,r,deep+1); // 同样是缩小区间,只不过变成了右边而已。注意要 k-cnt。 end; |
理论复杂度和亲测结果¶
时间复杂度 : 一次查询只需要
空间复杂度 : 只需要存储
亲测结果:主席树 :
后记¶
大家可以试着去写非递归版哦。参考博文 :传送门。
build本页面最近更新:,更新历史
edit发现错误?想一起完善? 在 GitHub 上编辑此页!
people本页面贡献者:
copyright本页面的全部内容在 CC BY-SA 4.0 和 SATA 协议之条款下提供,附加条款亦可能应用