栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

LeetCode 35. 搜索插入位置 | Python

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

LeetCode 35. 搜索插入位置 | Python

35. 搜索插入位置

题目来源:力扣(LeetCode)https://leetcode-cn.com/problems/search-insert-position

题目

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

你可以假设数组中无重复元素。

示例 1:

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

示例 2:

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

示例 3:

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

示例 4:

输入: [1,3,5,6], 0
输出: 0
解题思路

思路:二分查找

在这道题,题目中说明给定一个经过排序的数组,同时给定一个目标值。题目中的要求:

  • 如果目标值在数组中,返回其索引
  • 如果目标值不在数组中,返回其会被顺序插入的位置。

这里我们如果用暴力解的话,时间复杂度为 O(n)。但是要找有序数组中插入元素位置这种问题,我们可以考虑使用二分查找。

在这里,我们使用二分查找,遍历循环的过程中, 排除不符合条件的部分,最终剩下的就是答案。针对本题,可能出现的情况:

  • 因为目标值不一定存在于数组中,而且有可能目标值会被顺序插入到末尾。这里可以直接单独进行判断,返回数组的长度(也就是数组最末尾元素的下一个位置,如示例 3)
  • 通过观察示例,我们可以发现,当数组中的元素值严格小于目标值,那么它一定不是解。

二分查找使用相对简单,但是比较容易出错的地方在于边界的约束(需要注意)。

具体的代码实现如下(含注释)。

代码实现
class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
 length = len(nums)
 # 这里不单独判断目标值可能会被插入末尾的情况,统一进行判断
 # 设定边界
 left = 0
 # 因为目标值可能会被插入末尾,所以右边界设为 length
 right = length

 while left < right:
     # 开始进行二分查找
     # 这里是向下取整,mid 是包含在 [left, mid] 中的
     mid = left + (right - left) // 2
     # 严格小于 target 的元素,一定不是解
     if nums[mid] < target:
  left = mid + 1
     elif nums[mid] == target:
  # 当数组存在与目标值相同的元素时,返回其索引
  return mid
     else:
  right = mid
 
 return left
 
实现结果


转载请注明:文章转载自 www.mshxw.com
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号