刷题笔记-35. 搜索插入位置

113 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第1天,点击查看活动详情

一、题目描述:

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

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

请必须使用时间复杂度为 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分别记录列表两边的索引

然后判断左边的索引是否小于等于右边的索引,如果是的话就先计算出中间索引midpoint,中间索引由左右两边的索引整除2计算。

然后有三种情况,第一种情况如果target小于中间索引在nums中的数的话,表明target在中间索引的左边,所以将右边索引值设置为中间索引-1。

第二种情况,如果target大于中间索引在nums中的数的话,表面target在中间索引的右边,那么左边索引设置为中间索引+1.

第三种情况就是target等于中间索引在nums中的值,自然返回中间索引

那么如果nums中没有找到target的话,就返回左边索引。

三、AC 代码:

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


范文参考:

执行用时: 0 ms , 在所有 Java 提交中击败了 100.00% 的用户 内存消耗: 37.8 MB , 在所有 Java 提交中击败了 90.23% 的用户 - 搜索插入位置 - 力扣(LeetCode)