二分查找一般应用于在一个单调有序数组中快速定位目标元素的问题。比较基础,建议手撕背掉。
废话少说,直接步入正题。
【注:本文全部以单调非减的数组为例】
1、定位准确位置
Q1:给定数组 nums[n] 和目标值 target,求 target 在 nums[] 中的准确位置。
// 给定数组 nums[n] 和目标值 target,求 target 在 nums[] 中的准确位置。
int binarySearch(int[] nums, int target) {
if(nums.length == 0) return -1;
int left = 0, right = nums.length - 1; // Point 1
while(left <= right) { // Point 2
int mid = left + (right - left) / 2;
if(nums[mid] == target) {
return mid;
} else if(nums[mid] < target) {
left = mid + 1; // Point 3
} else if(nums[mid] > target) {
right = mid -1; // Point 4
}
}
return -1;
}
注意:
Point 2 处的 <=(或是 <)是由 Point 1 处的区间开闭来决定的。
若 [0, n-1],则对应 <=;
若 [0, n),则对应 <,此时要注意,Point 4 处应该为 right = mid。
Q2:给定数组 nums[n] 和目标值 target,求 target 在 nums[] 中的插入位置。
// 给定数组 nums[n] 和目标值 target,求 target 在 nums[] 中的插入位置。
int binarySearch(int[] nums, int target) {
if(nums.length == 0) return 0;
int left = 0, right = nums.length - 1; // Point 1
while(left <= right) { // Point 2
int mid = left + (right - left) / 2;
if(nums[mid] == target) {
return mid;
} else if(nums[mid] < target) {
left = mid + 1; // Point 3
} else if(nums[mid] > target) {
right = mid -1; // Point 4
}
}
return right + 1; // 注意,这里的不同
}
2、寻找右侧边界
Q3:给定数组 nums[n] 和目标值 target,求 target 在 nums[] 中的最大位置(最右侧位置)。
// 给定数组 nums[n] 和目标值 target,求 target 在 nums[] 中的最大位置。
int binarySearch(int[] nums, int target) {
if(nums.length == 0) return -1;
int left = 0, right = nums.length - 1; // Point 1
while(left <= right) { // Point 2
int mid = left + (right - left) / 2;
if(nums[mid] == target) {
left = mid + 1; // 不要返回,锁定右侧边界,
} else if(nums[mid] < target) {
left = mid + 1; // Point 3
} else if(nums[mid] > target) {
right = mid -1; // Point 4
}
}
if (right < 0 || nums[right] != target) {
return -1;
}
return right;
}
3、寻找左侧边界
Q4:给定数组 nums[n] 和目标值 target,求 target 在 nums[] 中的最小位置(最左侧位置)。
// 给定数组 nums[n] 和目标值 target,求 target 在 nums[] 中的最小位置。
int binarySearch(int[] nums, int target) {
if(nums.length == 0) return -1;
int left = 0, right = nums.length - 1; // Point 1
while(left <= right) { // Point 2
int mid = left + (right - left) / 2;
if(nums[mid] == target) {
right= mid - 1; // 不要返回,锁定右侧边界,
} else if(nums[mid] < target) {
left = mid + 1; // Point 3
} else if(nums[mid] > target) {
right = mid -1; // Point 4
}
}
if (left >= nums.length || nums[left] != target) {
return -1;
}
return left;
}