[路飞]_leetcode-35-搜索插入位置

336 阅读1分钟

「这是我参与11月更文挑战的第4天,活动详情查看:2021最后一次更文挑战

[题目地址] [B站地址]

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:

输入: nums = [1,3,5,6], target = 7
输出: 4

示例 4:

输入: nums = [1,3,5,6], target = 0
输出: 0

示例 5:

输入: nums = [1], target = 0
输出: 0

提示:

  • 1 <= nums.length <= 104
  • nums 为无重复元素升序排列数组
  • -104 <= target <= 104

本题就是让我们找到第一个大于等于目标值的位置即可。

针对这种问题,很自然的就想到可用使用 二分查找 来解题

二分查找

二分查找适用于在有序数组中找到第一个符合条件的位置

整体方法就是每次将搜索区间的中间值和目标值比较,通过大小关系判断目标值在当前区间的前半部分还是后半部分,然后将区间调整为对应部分,如此每次将查找区间减半,直到找到符合条件的元素或位置

至此,我们本题的解题思路就完成了,以下是代码实现

var searchInsert = function(nums, target) {
    if(target<nums[0]) return 0;
    if(target>nums[nums.length-1]) return nums.length;
    let l = 0,r = nums.length-1;
    while(l<r){
        const mid = (l+r)>>1;
        if(nums[mid]<target) l = mid+1
        else r = mid;
    }
    return l;
};

至此,我们就完成了leetcode-35-搜索插入位置

如有任何问题或建议,欢迎留言讨论!