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

示例 4:

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

示例 5:

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

提示:

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

1 <= len(nums) <= 10^4
-10^4 <= nums[i]、target <= 10^4
nums 为无重复元素的升序排列数组

分析

二分查找算法：复杂度O(logN)

题解

python3:（二分查找）

class Solution:
	def searchInsert(self, nums: List[int], target: int) -> int:
		left = 0
		right = len(nums)
		while (left < right):
			mid = left + (right - left) // 2
			if nums[mid] >= target:
				right = mid
			else:
				left = mid + 1
		return left