C语言
34. 在排序数组中查找元素的第一个和最后一个位置
在给定的官方解决代码中有函数 binarySearch:
int binarySearch(int* nums, int numsSize, int target, bool lower) {
int left = 0, right = numsSize - 1, ans = numsSize;
while (left <= right) {
int mid = (left + right) / 2;
if (nums[mid] > target || (lower && nums[mid] >= target))
{
right = mid - 1;
ans = mid;
} else {
left = mid + 1;
}
}
return ans;
}
而主要函数中有查找所求区间的左端点和右端点的代码:
int leftIdx = binarySearch(nums, numsSize, target, true);
int rightIdx = binarySearch(nums, numsSize, target, false) - 1;
①如果 lower == true,则在判断语句中将会等价于:
if (nums[mid] >= target){
right = mid - 1;
ans = mid;
}
else{
// nums[mid] < target
left = mid + 1;
}
若中间位置的值比目标值要大,则会查找左区间,当找到值与 target 相等的 mid 位置时并不会停止向左寻找,right 还是等于 mid - 1,因此最终找到的是区间左端点。
②如果 lower == false,则在在判断语句中将会是:
if (nums[mid] > target){
right = mid - 1;
ans = mid;
}
else {
// nums[mid] <= target
left = mid + 1;
}
此时的左端点确定无疑是在左侧区间,当找到值与 target 相等的 mid 位置时就将选择权交给else,即不会停止向右寻找,left 等于 mid + 1,但是由于ans是在上面被赋值的,所以 ans 最终会是区间右端点的右边临近值,因此最终的右端点需要 ans - 1 。
【right = mid - 1,当 left = mid + 1 > right 时,此时的 right 就是区间右端点的所在之处,而ans = mid】



