刷着玩-LeetCode20天算法-35. 搜索插入位置

35. 搜索插入位置

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

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

示例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

示例4：

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

示例5：

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

提示：

$1<=nums.length<=10^4$
$-10^4<=nums[i]<=10^4$
$nums为无重复元素的升序排列数组$
$-10^4<=target<=10^4$

题解：

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