给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
提示:
1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 为无重复元素的升序排列数组
-104 <= target <= 104
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
def midfind(left, right):
if left == right and nums[right] != target:
return right
mid = (left + right) // 2
if nums[mid] < target:
ans = midfind(mid + 1, right)
else:
if nums[mid] == target:
return mid
else:
ans = midfind(left, mid)
return ans
ans=midfind(0,len(nums)-1)
if ans==len(nums)-1 and nums[-1]<target:
return len(nums)
return ans
思路:把昨天的代码改改,又是一条好汉!