力扣每日一题35：搜索插入的位置

暴力递归战士2023-10-12 9:11

题目描述：

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

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

`思路及题解：`

找到第一个大于等于target的数的下标即可，如果没有大于等于target的数，就插入数组尾部，即返回numsSize。用二分法查找。

代码：

int searchInsert(int* nums, int numsSize, int target){
    int lo=0;
    int hi=numsSize-1;
    int mid;
    int ans=numsSize;
    // if(nums[0]>target) return 0;
    // if(target>nums[hi]) return numsSize;
    //找第一个大于等于target的数
    while(lo<=hi)
    {
        mid=(lo+hi)/2;
        if(target<=nums[mid])
        {
            ans=mid;
            hi=mid-1;
        }
        else
        {
            lo=mid+1;
        }
    }
    return ans;
}