【leetcode】35. 搜索插入位置

题目

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. 一次遍历

class Solution:def searchInsert(self, nums: List[int], target: int) -> int:i = 0n = len(nums)while i < n:if nums[i] < target:i += 1elif nums[i] == target:return ielse:return ireturn i

2. 库函数

class Solution:def searchInsert(self, nums: List[int], target: int) -> int:return bisect_left(nums, target)

3. 二分

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