二分查找3️⃣-搜索插入位置-CSDN博客

本文链接：https://blog.csdn.net/2202_75331338/article/details/141059989

题目描述：

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

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

1 <= nums.length <= 104

-104 <= nums[i] <= 104

nums 为 无重复元素 的升序排列数组

-104 <= target <= 104

解法（二分查找算法）：

算法思路：

a. 分析插入位置左右两侧区间上元素的特点：

设插入位置的坐标为 index ，根据插入位置的特点可以知道：

• [left, index - 1] 内的所有元素均是小于 target 的；

• [index, right] 内的所有元素均是大于等于 target 的。

b. 设 left 为本轮查询的左边界， right 为本轮查询的右边界。根据mid 位置元素的信息，分析下一轮查询的区间：

▪ 当 nums[mid] >= target 时，说明 mid 落在了[index, right] 区间上，mid 左边包括 mid 本身，可能是最终结果，所以我们接下来查找的区间在 [left,mid] 上。因此，更新 right 到 mid 位置，继续查找。

▪ 当 nums[mid] < target 时，说明 mid 落在了[left, index - 1]区间上，mid 右边但不包括 mid 本身，可能是最终结果，所以我们接下来查找的区间在[mid + 1, right]上。因此，更新left 到mid + 1点位置，继续查找。

c. 直到我们的查找区间的长度变为 1 ，也就是 left == right 的时候， left 或者right 所在的位置就是我们要找的结果。

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