leetcode 链接：35. 搜索插入位置

题目

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

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

示例：

输入: nums = [1,3,5,6], target = 5
输出: 2

输入: nums = [1,3,5,6], target = 2
输出: 1

输入: nums = [1,3,5,6], target = 7
输出: 4

输入: nums = [1,3,5,6], target = 0
输出: 0

输入: nums = [1], target = 0
输出: 0

解答 & 代码

二分查找
在左闭右闭区间 [left, right] 搜索 target，因此循环条件是 while(left <= right) ，因为 left == right 时搜索范围不为空

• 若 nums[mid] > target，说明 target 在左区间，不包含 mid，因此 right = mid - 1

最终若没找到 target，则此时 left 的位置（即 right + 1）的位置，就是要插入的位置

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;    // 在左闭右闭区间 [left, right] 搜索 target
        while(left <= right)            // 当left==right，区间[left, right]依然有效
        {
            int mid = left + (right - left) / 2;    // 防止溢出
            if(nums[mid] == target)
                return mid;                // 找到 target，返回对应位置
            else if(nums[mid] > target)
                right = mid - 1;        // target 在左区间，所以[left, middle - 1]
            else
                left = mid + 1;            // target 在右区间，所以[middle + 1, right]
        }
        return left;    // or return right + 1
    }
};

执行结果：

执行结果：通过

执行用时：0 ms, 在所有 C++ 提交中击败了 100.00% 的用户
内存消耗：9.5 MB, 在所有 C++ 提交中击败了 5.96% 的用户