35. 搜索插入位置

最新推荐文章于 2024-08-10 23:59:21 发布

Karynnn

最新推荐文章于 2024-08-10 23:59:21 发布

阅读量66

点赞数

分类专栏： Code 文章标签： leetcode 二分查找 c++

本文链接：https://blog.csdn.net/qq_50920130/article/details/118569705

版权

Code 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

35. 搜索插入位置

题目：
思路：
代码：
Debug:
相关题目

题目：

原题链接

在这里插入图片描述

思路：

按照二分查找的思路来，然后调用vector的insert就可以。

代码：

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        int middle = (left + right) / 2;

        while (left <= right) {
            if (nums[middle] == target) {
                return middle;
            }
            else if (nums[middle] < target) {
                left = middle + 1;
                middle = (left + right) / 2;
            }
            else {
                right = middle - 1;
                middle = (left + right) / 2;
            }
        }
        nums.insert(nums.begin() + middle, target);
        return right+1;
    }
};

Debug:

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        int middle = (left + right) / 2;
		// 二分查找
        while (left <= right) {
            if (nums[middle] == target) {
                // 返回找到target的位置
                return middle;
            }
            else if (nums[middle] < target) {
                left = middle + 1;
                middle = (left + right) / 2;
            }
            else {
                right = middle - 1;
                middle = (left + right) / 2;
            }
        }
        // 没找到，就在合适的位置insert
        nums.insert(nums.begin() + middle, target);
        // 返回插入的位置
        return middle;
    }
};

第一次提交的代码指通过了39/62个测试用例，如下图

在这里插入图片描述

从该测试用例来看，程序执行过程如下：

left = 0, right = 3, middle = 1, val = 2
nums[middle] = 3 > val
right = middle -1 = 0, left = 0, middle = 0
nums[middle] = 1 < val
left = middle +1 = 1, right = 0, middle = 0, left > right 跳出循环
本代码插入的位置是0，正确答案应该是1。

经过分析，如果最后跳出循环，那么必然是因为left > right，反推一下，在循环中发生了left = middle + 1或者right = middle - 1。
left = middle + 1：说明nums[middle] < target，那么要插入的位置就是当前middle + 1。但是因为会执行第15行的middle = (left + right) / 2，所以需要插入的是middle+1或right+1位置。
right = middle - 1：说明nums[middle] > target，那么要插入的位置就是当前middle - 1。因为middle = (left + right)/2仍要执行，所以需要插入的是middle+1或right+1位置。

第二次提交，尝试return middle+1，出现以下错误。