search insert position
Given a sorted array of distinct integers 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.
如果找到这个value,则返回index,如果没有找到,返回按照顺序被插入的index.
常规法
第一种 一个麻烦的解法
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
n=len(nums)
for i in range(n-1):
if nums[i]==target:
return i
if nums[i]<target and nums[i+1]>target:
return i+1
if nums[n-1]==target:
return n-1
if nums[n-1]<target:
return n
if nums[0]>target:
return 0
这个就是考虑了各种各样的边界条件写的比较笨的算法
第二种
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
if nums==None or len(nums)==0:
return 0
for i,num in enumerate(nums):
if num>=target:
return i
return len(nums)
(1)相比较于上个算法,这个算法比较简洁。主要是采用了enumerate
(2)第一行,首先是if nums==None or len(mums)==0,是说如果list是空的,那么就直接插入到index=0位置就行了
(3)最后一个return len(nums), 也属于边界条件。就是当所有遍历之后都没有,就返回索引len(nums)
二分法
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
if nums==None or len(nums)==0:
return 0
left=0
right=len(nums)-1
while left<right:
mid=left+(right-left)//2
if nums[mid]==target:
return mid
elif nums[mid]>target:
right=mid
else:
left=mid+1
return left if nums[left]>=target else left+1
(1)要定义两个指针,一个指向头部0,一个指向尾部len-1,别忘了减1的事情
(2)下面就是二分法了,需要记住