LeetCode 35.搜索插入位置

LeetCode 35.搜索插入位置

1、题目

题目链接:35. 搜索插入位置

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

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

示例 1:

markdown 复制代码
输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

markdown 复制代码
输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:

markdown 复制代码
输入: nums = [1,3,5,6], target = 7
输出: 4

提示:

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

2、思路

题目要求我们找到 目标值有序数组中 会被 顺序插入的位置,其实插入的位置无非是以下四种情况。

  • 目标值在数组所有元素之前
  • 目标值等于数组中某一个元素
  • 目标值插入数组中的位置
  • 目标值在数组所有元素之后

根据题目可知,数组是有序的,在 有序数组 中查找符合条件的某个数(或者它的下标),这是二分查找的基本条件。同时题目又强调数组中 无重复元素,这样使用二分查找返回的元素下标就是唯一的。所以这道题可以使用二分查找来求解。

3、二分法(左闭右闭区间)

代码:

cpp 复制代码
class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int n = nums.size();
        int left = 0;
        int right = n - 1; // 定义target在左闭右闭的区间里,[left, right]
        while (left <= right) { // 当 left == right,区间 [left, right]依然有效
            int middle = left + ((right - left) / 2);// 防止溢出 等同于(left + right)/2
            if (nums[middle] > target) {
                right = middle - 1; // target 在左区间,所以[left, middle - 1]
            } else if (nums[middle] < target) {
                left = middle + 1; // target 在右区间,所以[middle + 1, right]
            } else { // nums[middle] == target
                return middle;
            }
        }
        // 分别处理如下四种情况
        // 目标值在数组所有元素之前  [0, -1]
        // 目标值等于数组中某一个元素  return middle;
        // 目标值插入数组中的位置 [left, right],return  right + 1
        // 目标值在数组所有元素之后的情况 [left, right], 因为是右闭区间,所以 return right + 1
        
        return right + 1;
        // return left;
    }
};

复杂度分析

  • 时间复杂度:O(log n)
  • 空间复杂度:O(1)

4、二分法(左闭右开区间)

代码:

cpp 复制代码
class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size(); // 定义target在左闭右开的区间里,[left, right)

        while (left < right) { // 因为left == right的时候,在[left, right)是无效的空间
            int middle = left + ((right - left) >> 1);

            if (nums[middle] > target) {
                right = middle; // target 在左区间,在[left, middle)中
            } else if (nums[middle] < target) {
                left = middle + 1; // target 在右区间,在 [middle + 1, right)中
            } else { // nums[middle] == target
                return middle; // 数组中找到目标值的情况,直接返回下标
            }
        }
        // 分别处理如下四种情况
        // 目标值在数组所有元素之前 [0,0)
        // 目标值等于数组中某一个元素 return middle
        // 目标值插入数组中的位置 [left, right) ,return right 即可
        // 目标值在数组所有元素之后的情况 [left, right),因为是右开区间,所以 return right
        return right;
        // return left
    }
};

复杂度分析

  • 时间复杂度:O(log n)
  • 空间复杂度:O(1)
相关推荐
XXX-X-XXJ11 分钟前
三、从 MinIO 存储到 OCR 提取,再到向量索引生成
人工智能·后端·python·ocr
该用户已不存在19 分钟前
7个没听过但很好用的Mac工具
后端·开源
码事漫谈22 分钟前
如何设置 Visual Studio 在调试停止时自动关闭控制台
后端
码事漫谈24 分钟前
函数封装的平衡艺术:以C++为例探讨适度封装
后端
程序员东岸25 分钟前
学完顺序表后,用 C 语言写了一个通讯录
数据结构·笔记·学习
hello 早上好35 分钟前
深入 Spring 条件化配置底层:从硬编码到通用注解的实现原理
java·后端·spring
亚林瓜子41 分钟前
Spring中Date日期序列化与反序列化中格式设置
java·后端·spring·jackson·date
棉猴1 小时前
GESP C++等级认证三级15-原码反码补码2-2
开发语言·c++·gesp·c++三级·等级认证·原码反码补码
IT古董1 小时前
【第五章:计算机视觉-项目实战之生成式算法实战:扩散模型】3.生成式算法实战:扩散模型-(3)DDPM模型训练与推理
人工智能·算法·计算机视觉
独自破碎E1 小时前
Leetcode2166-设计位集
java·数据结构·算法