LeetCode 35. 搜索插入位置 (C++实现)

姚先生972024-12-22 23:00

1. 题目描述

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

请必须使用时间复杂度为 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

2. 解题思路

首先定义左指针left和右指针right，然后取数组中间元素的值和target比较，如果中间元素的值大于target，则右指针指向中间元素，如果中间元素的值小于target，则左指针指向中间元素，直到中间元素等于target。最后，如果没有找到，则右指针的下一位就是插入的位置。

3. 代码实现

cpp 复制代码

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