对权值线段树剪枝的误解--以HDU6703为例
引子
对hdu6703,首先将问题转化为“询问一个排列中大于等于k的值里,下标超过r的最小权值是多少”
我们采用官方题解中的做法:权值线段树+剪枝
对(a[i],i)建线段树,查询权值线段树的[k,n]中第一个下标超过r的值
代码是这样的
int ask(int l, int r, int root){
int mid = (l+r)>>1;
if(mx[root]<=R)return -1;
if(l==r){return l;}
int ans = -1;
if(k<=mid)ans=ask(lson);
if(ans==-1)ans=ask(rson);
return ans;
}
这把辣鸡的我给看meng了:在R=n的时候,最坏情况下一直向左递归,并且没找到然后向右递归,再向右递归的同时又重复没找到,这个ask不就退化成O(n)的了吗?
思考
经过半个月的思考(被虐),我大概懂了这个剪枝
首先分析以下几个问题:
什么情况下才会递归下去?
由代码第三行的
if(mx[root]<=R)return -1;
我们可以知道,只有当前权值区间\((l,r)\)的最大下标超过R才可能存在答案
什么情况下会往左递归并且不会从左边的递归返回答案?
假设当前权值区间为\([l,r]\)
如果往左边递归没有O(1)返回的话,根据上面的结论,那么一定是因为左区间\([l,mid]\)存在一个下标大于R
但是,左区间中合法区间应该为\([max(k,l),mid]\) **
所以当\(k>l\),且答案均分布在\([l,k]\)时,才会向左递归并且不从左区间返回答案**
什么时候“错误的左区间递归”会结束?
假设最坏情况,答案在k-1里,k-1一直在做区间的递归中,只有递归到当l=k的时候,才会结束这个错误
由线段树的相关性质只可以知道,这个最坏情况可以到\(l=r=k\),也就是跑了一个\(O(logn)\)的链
深入思考
思考完以上几个问题,继续思考:
当走完这条错误链,回溯的时候,会回溯到哪里?
当然是第一次出现这个错误分叉的地方(其实就是父节点)
但是此时我们在左区间没找到答案,会去右区间,而右区间\([mid+1,r]\)是完全包含于合法区间\([k,n]\)中的,所以只会出现两种情况
1.右区间没有合法答案,O(1)退出,继续回溯
2.右区间有答案,最终答案必在右区间中
一旦出现了2,就是正常的没有限制\([k,n]\)的线段树找最小值的O(logn)的做法了
而1也只是一个普通的回溯,按照父节点回溯到最原始的错误分叉,答案就在另一条路中
这个问题就解决啦
结论
这个剪枝强无敌,最终询问操作的执行次数只有两条链
复杂度为O(logn)