题目:
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-search
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
我的代码:
class Solution {
public:
int search(vector<int>& nums, int target) {
int n=nums.size();
for(int i=0; i<n; i++){
if(nums[n-1] >= target){
if(nums[i] == target)
return i;
else if(nums[i] > target)
return -1;
}
else
return -1;
}
return 0;
}
};
笔记:链接(力扣)
二分查找常用场景:寻找一个数、寻找左侧边界、寻找右侧边界。
二分查找框架:
int binarySearch(int[] nums, int target) {
int left = 0, right = ...;
while(...) {
int mid = left + (right - left) / 2;
if (nums[mid] == target) {
...
} else if (nums[mid] < target) {
left = ...
} else if (nums[mid] > target) {
right = ...
}
}
return ...;
}
注1:写清楚所有的情况,使用 else if 而不仅仅是else。
注2:计算 mid 时需要防止溢出。left + (right - left) / 2 的结果和 (left + right) / 2 结果一样,但有效防止了 left
和 right
太大直接相加导致溢出。
一、寻找一个数(基本的二分查找)
int binarySearch(int[] nums, int target) {
int left = 0;
int right = nums.length - 1; //数组下标从0开始
while(left <= right) {
int mid = left + (right - left) / 2;
if(nums[mid] == target) {
return mid;
else if(nums[mid] < target)
left = mid + 1;
else if(nums[mid] > target)
right = mid - 1;
}
return -1;
}
二、寻找左侧边界的二分查找
int left_bound(int[] nums, int target) {
int left = 0;
int right = nums.length - 1; //数组下标从0开始
//搜索区间为[left, right]
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最终等于right,然后统一返回left值
right = mid - 1;
}
//检查出界情况
if(left >= nums.length || nums[left] != target) //target大于or小于所有的数,索引越界
return -1;
return left;
}
三、寻找右侧边界的二分查找
int right_bound(int[] nums, int target) {
int left = 0;
int 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;
}
if(right < 0 || nums[right] != target) //target小于or大于所有元素
return -1;
}