LeetCode 35. 搜索插入位置

dreamer'~2023-12-08 14:00

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
提示:

1 <= nums.length <= 104

-104 <= nums[i] <= 104

nums 为 无重复元素 的升序排列数组

-104 <= target <= 104

思路：与普通的二分查找类似，只不过对于target不存在的情况，需要返回left

时间复杂度：O(logn)

空间复杂度： O(1)

Go 复制代码

// 参考视频题解：https://leetcode.cn/problems/search-insert-position/solutions/333632/sou-suo-cha-ru-wei-zhi-by-leetcode-solution/?envType=study-plan-v2&envId=top-100-liked
// 与普通的二分查找类似，只不过对于target不存在的情况，需要返回left
func searchInsert(nums []int, target int) int {
    left, right := 0, len(nums)-1
    
    for left <= right {
        mid := left + (right-left)/2
        if target == nums[mid] {
            return mid
        } else if target > nums[mid] {
            left = mid+1
        } else {
            right = mid-1
        }
    }

    return left
}