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

题目要求必须使用时间复杂度为O(log n) 的算法。ban了最简单的遍历。

int searchInsert(int* nums, int numsSize, int target){
    for(int i=0;i<numsSize;i++){
        if(nums[i]>=target)
           return i;
    }
    return numsSize;
}

而上一道题的二分查找时间复杂度就是O(log n)

采用二分查找

int searchInsert(int* nums, int numsSize, int target){
    int left=0,right=numsSize-1,middle;
    while(left<=right){
        middle=(left+right)/2;
        if(nums[middle]>target){
            right=middle-1;
        }else if(nums[middle]<target){
            left=middle+1;
        }else return middle;
    }
    return right+1;
}

注释

 //[left, right]左闭右闭区间
int searchInsert(int* nums, int numsSize, int target){
    //左闭右开区间 [0 , numsSize-1]
        int left =0;
        int mid =0;
        int right = numsSize - 1;
        while(left <= right){//左闭右闭区间 所以可以 left == right
            mid = left + (right - left) / 2;
            if(target < nums[mid]){
        //target 在左区间 [left, mid - 1]中，原区间包含mid，右区间边界可以向左内缩
                right = mid -1;
            }else if( target > nums[mid]){
        //target 在右区间 [mid + 1, right]中,原区间包含mid，左区间边界可以向右内缩
                left = mid + 1;
            }else {
        // nums[mid] == target ，顺利找到target，直接返回mid
                return mid;
            }
        }
        //数组中未找到target元素
        //target在数组所有元素之后，[left, right]是右闭区间，需要返回 right +1
        return right + 1;
}