二分查找
704.二分查找
题目简介
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
例如:
输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4
思路
定义的 target 是在一个在左闭右闭的区间里,也就是[left, right] 。区间的定义决定了二分法的代码应该如何写,因为定义target在[left, right]区间,所以有如下两点:
- while (left <= right) 要使用 <= ,因为left == right是有意义的,所以使用 <=
- if (nums[middle] > target) right 要赋值为 middle - 1,因为当前这个nums[middle]一定不是target,那么接下来要查找的左区间结束下标位置就是 middle - 1
- 同理 if (nums[middle] < target) left 要赋值为 middle + 1
代码实现
function search(nums: number[], target: number): number {
// 边界[left,right](左闭右闭)
let [l,r]=[0,nums.length-1]
// 当left==right,区间[left, right]依然有效,所以用 <=
while(l<=r){
// 防止溢出 等同于(left + right)/2
const mid=Math.floor((r-l)/2+l)
// target在右区间,继续在[mid+1,right]中找target
if(nums[mid]<target) l=mid+1
// target在左区间,继续在[left,mid-1]中找target
else if(nums[mid]>target) r=mid-1
// 找到target返回下标
else return mid
}
// 未找到,返回-1
return -1
};
相关题目推荐
35.搜索插入位置
function searchInsert(nums: number[], target: number): number {
// 边界[left,right](左闭右闭)
let [l,r]=[0,nums.length-1]
// 当left==right,区间[left, right]依然有效,所以用 <=
while(l<=r){
// 防止溢出 等同于(left + right)/2
const mid=Math.floor((r-l)/2+l)
// 找到target返回下标
if(nums[mid]===target) return mid
// target在右区间,继续在[mid+1,right]中找target
else if(nums[mid]<target) l=mid+1
// target在左区间,继续在[left,mid-1]中找target
else r=mid-1
}
// 未在数组中找到,此时r为所有小于target值中最大值的下标
// 所以为保证数组有序,在最大值的右边插入target
return r+1
};
34. 在排序数组中查找元素的第一个和最后一个位置
function searchRange(nums: number[], target: number): number[] {
// 边界[left,right](左闭右闭)
let [l,r]=[0,nums.length-1]
// 当left==right,区间[left, right]依然有效,所以用 <=
while(l<=r){
// 防止溢出 等同于(left + right)/2
const mid=Math.floor((r-l)/2+l)
// target在右区间,继续在[mid+1,right]中找target
if(nums[mid]<target) l=mid+1
// target在左区间,继续在[left,mid-1]中找target
if(nums[mid]>target) r=mid-1
// 找到target,判断左右边界是否为target
if(nums[mid]===target){
// 若不是,则缩小左右边界
if(nums[r]!==target) r-=1
if(nums[l]!==target) l+=1
}
// 若左右边界都是target,则返回下标
if(nums[l]===target&&nums[r]===target) return [l,r]
}
// 未找到
return [-1,-1]
};
69. x 的平方根
同样利用二分的思想,在【0,x】范围内找到一个数的平方最接近X
function mySqrt(x: number): number {
// // 边界[left,right](左闭右闭)
let [l,r]=[0,x]
// 记录最接近的答案
let ans=0
while(l<=r){
// 防止溢出 等同于(left + right)/2
const mid=Math.floor((r-l)/2+l)
// 若中间值的平方<=X,记录中间值,继续在[mid+1,x]找是否有更接近x的值
if(mid*mid<=x) {
ans=mid
l=mid+1
}
// 中间值的平方>x,在[l,mid-1]中找
else r=mid-1
}
// 返回最接近的值
return ans
};
367. 有效的完全平方数
与 “69. x 的平方根” 相似,可以看作在[0,1,2,…,num-1,num]中找一个数的平方等于num
function isPerfectSquare(num: number): boolean {
let [l,r]=[0,num]
while(l<=r){
const mid=Math.floor((r-l)/2+l)
const sqr=mid*mid
if(sqr<num) l=mid+1
else if(sqr>num) r=mid-1
else return true
}
return false
};