前言:已经是一枚妥妥研二党了,科研之余不能不学学找工作的玩意了。
刷题的话,跟的是[代码随想录]的刷题顺序,先从数组篇入手,大佬用的是C++,我用的是python,为了让同用python的小伙伴们有答案可循,于是就想把自己的刷题经历和学习过程记录在小本本上。
先来看看题目:搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 1:
输入: [1,3,5,6], 5
输出: 2
示例 2:
输入: [1,3,5,6], 2
输出: 1
示例 3:
输入: [1,3,5,6], 7
输出: 4
示例 4:
输入: [1,3,5,6], 0
输出: 0
思路:
插入元素的位置有四种情况
1.目标在数组所有元素之前;
2.目标等于数组某一元素;
3.目标值插入数组中的位置;
4.目标值在数组所有元素之后。
代码1:暴力解法
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
for i in nums:
if i>=target:
return nums.index(i)
if nums[0]>=target:
return 0
else:
return len(nums)
代码2:二分法
暴力解法的时间复杂度高,此处再引入二分法的解法,降低了时间复杂度。
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
n=len(nums)
left=0
right=n-1
while left <= right:
middle = int(left+((right-left)/2))
if nums[middle]>target:
right=middle-1
elif nums[middle]<target:
left=middle+1
else:
return middle
return right+1
对于二分法而言,最重要的是左右指针的选取,具体的内容会和滑动窗口的解法一起说明的。