每日一题95——搜索插入位置

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

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

解题思路：

要求时间复杂度为 O(log n) 的算法，考虑二分查找。分为四种情况讨论，1.目标值在所有元素之前，2.目标值在所有元素之后，3.目标值是数组中的某一个元素，4.目标值需要插入数组中的某个位置。

 // 分别处理如下四种情况
        // 目标值在数组所有元素之前  [0, -1]
        // 目标值等于数组中某一个元素  return middle;
        // 目标值插入数组中的位置 [left, right]，return  right + 1
        // 目标值在数组所有元素之后的情况 [left, right]，这是右闭区间，所以  return right + 1

代码：

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int n = nums.size(),l,r,mid;
        l = 0,r = n - 1;
        //目标在数组所有元素之前
        if(nums[0] >= target) return 0;
        
        //目标在数组元素中
        while(l <= r){
            mid = (l + r) / 2;
            if(nums[mid] == target){
                return mid;
            }else if(nums[mid] > target){
                r = mid - 1;
            }else{
                l = mid + 1;
            }
        }
        //目标在数组所有元素之后或者需要插入
        return r + 1;
    }
};