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) ，明显是二分查找问题。

def searchInsert(nums, target):
	left, right = 0, len(nums)-1
	while left <= right:
		mid = (left + rgith) // 2
		if nums[mid] == target:
			return mid
		elif nums[mid] < target:
			left = mid + 1
		else:
			right = mid + 1
	else:
		return left

nums = [1,3,5,6]
target = 5
print(searchInsert(nums, target)