1.编辑器
我使用的是win10+vscode+leetcode+python3
环境配置参见我的博客:
链接
2.第三十五题
(1)题目
英文:
Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.
中文:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/valid-parentheses
(2)解法
① 先将target插入到nums中,然后去除重复的(如果没有重复的,就说明是要插入的元素了),然后转成list(因为目前是无序的,所以需要sorted),最后再索引出插入的index即可。
(耗时48ms,内存:14.8M)
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
return sorted(list(set(nums+[target]))).index(target)
② 二分法 (有序数组)
(耗时36ms,内存:14.1M)
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
low, high = 0, len(nums)-1
while (low<=high):
middle = low+(high-low)//2
if nums[middle]==target:
return middle
elif nums[middle]>target:
high = middle-1
else:
low = middle+1
return low
注意:
1.不要用/,这个是有小数的,而index是没有小数的。
③ 遍历(因为数组是有序的,可以依次尝试插值)
(耗时48ms,内存:14.2M)
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
i = 0
while i < len(nums):
if nums[i] < target:
i +=1
else:
break
return i