一,在数组中找一个与目标数相等的数
//在数组中找一个数
int binary_search(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while(left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] < target) {
left = mid + 1;
} else if (nums[mid] > target) {
right = mid - 1;
} else if(nums[mid] == target) {
// 直接返回
return mid;
}
}
// 直接返回
return -1;
}
二,找左右边界
//找左边界,注意左左越
int left_bound(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] < target) {
left = mid + 1;
} else if (nums[mid] > target) {
right = mid - 1;
} else if (nums[mid] == target) {
// 别返回,收缩左侧边界
right = mid - 1;
}
}
// 由于 while 的退出条件是 left == right + 1,所以当 target 比 nums 中所有元素都大时,会存在以下情况使得索引越界:
if (left >= nums.length || nums[left] != target)
return -1;
return left;
}
//找右边界,右不够
int right_bound(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] < target) {
left = mid + 1;
} else if (nums[mid] > target) {
right = mid - 1;
} else if (nums[mid] == target) {
// 别返回,收缩右侧边界
left = mid + 1;
}
}
// 最后要检查 right 越界的情况
if (right < 0 || nums[right] != target)
return -1;
return right;
}
特别要注意的点:
一:在求左边界的时候, 由于 while 的退出条件是 left == right + 1,所以当 target 比 nums 中所有元素都大时,会存在以下情况使得索引越界:图片链接为:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/solution/er-fen-cha-zhao-suan-fa-xi-jie-xiang-jie-by-labula/
因此需要加上这些代码:
if (left >= nums.length || nums[left] != target)
return -1;
return left;
第二点:在求右边界的时候,当 target
比所有元素都小时,right
会被减到 -1,所以需要在最后防止越界
需要加上这些代码:
if (right < 0 || nums[right] != target)
return -1;
return right;
第三点:
求左边界的时候,当找到数组中的值和目标值相等时,而不是直接返回,而是进一步缩短左边界的大小。
求右边界的时候,当找到数组中的值和目标值相等时,不是直接返回,而是进一步缩短有边界的大小。
} else if (nums[mid] == target) {
// 别返回,收缩左侧边界
right = mid - 1;
}
} else if (nums[mid] == target) {
// 别返回,收缩右侧边界
left = mid + 1;
}