持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第1天,点击查看活动详情
一、题目描述:
35. 搜索插入位置 - 力扣(LeetCode) (leetcode-cn.com)
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 的算法。
示例 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
- 首先判断特殊情况——target比数组里最后一个数字都要大,那么直接插入最后一个,即返回数组长度即可
- 正常情况下的判断,我们创建一个mid索引将数组分为两个区间,这里有三种情况:如果mid==target,直接返回 如果mid比target大,那说明mid后面的都比target大,令right=mid-1,如果mid<target,代表mid左边的都比target小,令left=mid+1
- 最后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