二分法通常涉及到一前一后的下标i和j,这个技术用在了很多算法上,比如找第一个大于x的数,快排等等等等。下标一不小心就容易出错,这里总结一些一些用法。
一、找第一个>=target的数
int i = 0, j = A.size() - 1, mid;
while(i < j){
mid = i + (j - i) / 2;
if(A[mid] < target){
i = mid + 1;
} else {
j = mid;
}
}
当A[mid]大于等于target的时候,j移动到mid的位置,也就是说,j的位置,永远都是大于等于target。但是当A[mid]小于target的时候,i移动到mid+1,i的大小性并不确定,有可能有小于target,等于target或者是大于target。跳出条件是i<j,也就是说,有可能从j=mid跳出,也可能从i=mid+1跳出。分析下面两种情况:
1、当从j=mid跳出的时候,显然此时i=mid,也就是说i-1要么是0,要么A[i-1] < target,所以这个时候A[j]是从左往右第一个>=target的数
2、当从i=mid+1跳出的时候,j要么是最后一位,要么A[j] >= target。
所以上面的代码就是找到第一个>=target的数,如果不存在,要么都小于target,则返回最后一个坐标,要么都大于target,返回第一个坐标。
另外,涉及到一个可退出性的问题。也就是说,i必须要比j大,才会退出。考虑i=j-1的情况,此时mid为i。所以对i的操作不能是i=mid,不然会死循环。