模板
bool check(int x) {/* ... */} // 检查x是否满足某种性质
// 区间[l, r]被划分成[l, mid]和[mid + 1, r]时使用:
int bsearch_1(int l, int r)
{
while (l < r)
{
int mid = l + r >> 1;
if (check(mid)) r = mid; // check()判断mid是否满足性质
else l = mid + 1;
}
return l;
}
// 区间[l, r]被划分成[l, mid - 1]和[mid, r]时使用:
int bsearch_2(int l, int r)
{
while (l < r)
{
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
return l;
}
作者:yxc
链接:https://www.acwing.com/blog/content/277/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
注意两点
1.mid取值问题:如何选择L+R与L+R+1。当L=mid时,mid=L+R+1>>1,防止陷入死循环。当R=mid时,mid=L+R>>1。
分析方法:当L与R仅仅相差1时,L=mid=L+R>>1=L,此时则会陷入死循环,所以应让mid=L+R+1>>1。另一种情况同理。
2.二分结果:二分结束在while(L<R),结束时L==R。
题目一
// Forward declaration of compare API.
// bool compare(int a, int b);
// return bool means whether a is less than b.
class Solution {
public:
vector<int> specialSort(int N) {
vector<int> res;
res.push_back(1);
for(int i=2;i<=N;i++)
{
int l=0,r=res.size()-1;
while(l<r)
{
int mid=l+r+1>>1;
if(compare(i,res[mid])) r=mid-1;
else l=mid;
}
res.push_back(i);
//for(int j=res.size()-1;j>=r;j--) swap(res[j],i);
for(int j=res.size()-2;j>r;j--) swap(res[j],res[j+1]);
if(compare(i,res[r])) swap(res[0],res[1]);
}
return res;
}
};
边界问题:
最后插入答案的时候,答案插入的位置可能是第一个之前或者最后一个之后,但是二分结果只能在第一个和最后一个之间,不能显示第一个之前和最后一个之后,所以需要特殊判断,而我们插入时总是插到最后一个,再不断往前移,所以只需考虑第一个之前的情况。也就是最后一步的特判。