35.搜索插入位置

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

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

思路

这道题应该用二分法 但是插入位置要慎重考虑

插入位置有三种情况：对于插到数组前后的两种特殊情况应该单独分类讨论

代码

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        if(target < nums[0]) return 0;//对于插入首尾的两种特殊情况要单独考虑
        if(target > nums[nums.size() - 1]) return nums.size();
        
        int left = 0;
        int right = nums.size() - 1;
        int mid = 0;

        while(left <= right){
            mid = (left + right) / 2;
            if(nums[mid] == target){
                return mid;
            }else if(nums[mid] < target){
                left = mid + 1;
            }else{
                right = mid - 1;
            }
        }   
        return left;
    }
};