算法笔记 -- 35. 搜索插入位置

73 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第1天,点击查看活动详情

一、题目描述:

35. 搜索插入位置 - 力扣(LeetCode) (leetcode-cn.com)

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

请必须使用时间复杂度为O(logn) 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 <= 10^4
  • -10^4 <= nums[i] <= 10^4
  • nums 为无重复元素的升序排列数组
  • -10^4 <= target <= 10^4

二、思路分析:

经典二分查找法,首先确定左右边界left和right

  1. 首先判断特殊情况——target比数组里最后一个数字都要大,那么直接插入最后一个,即返回数组长度即可
  2. 正常情况下的判断,我们创建一个mid索引将数组分为两个区间,这里有三种情况:如果mid==target,直接返回 如果mid比target大,那说明mid后面的都比target大,令right=mid-1,如果mid<target,代表mid左边的都比target小,令left=mid+1
  3. 最后return left即可,因为就算target比数组里第一个数字都要小也没关系,我们知道这时候需要返回0,这种情况下是right指针会一直向左移动,left不会改变,而left恰好就是0

三、AC 代码:

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        if nums[len(nums)-1]<target:return len(nums)
        left,right=0,len(nums)-1
        while left<=right:
            mid=left+(right-left)//2
            if nums[mid]==target:return mid
            elif nums[mid]>target:
                right=mid-1
            else:
                left=mid+1
        return left

四、:参考 简单易懂,第一步先判断是否target是否在其中,如果在的话,程序肯定执行到return i,返回所在i的值,如果不在,就直接执行下面一个循环,进行比较大小,即可,最终返回count的数值,即使target在顺序中相对应的位置。 - 搜索插入位置 - 力扣(LeetCode)