《代码随想录》Day2
1. Q977. 有序数组的平方
给定一个非递减顺序的整数数组nums,返回由该数组每个数字的平方组成的新数组,且也为非递减顺序
暴力解法
直接平方后用sorted函数,但时间复杂度比较高
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
return sorted(num*num for num in nums)
双指针法:两端向中间找
利用数组有序(非递减顺序)的条件,原数组平方的最大值只能来自原数组的两端(考虑负数平方后可能为最大值的情况),则可利用双指针,a从数组前端开始,b从数组后端开始,比较出较大值后放入新数组末尾空位。
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
n = len(nums)
a, b, i = 0, n-1, n-1
ans = [0] * n #由于是从后往前依次填充,所以需要先把list结构和大小先定义好
while a <= b : #可等于
if nums[a]*nums[a] <= nums[b]*nums[b]: #则先将b放入新数组
ans[i] = nums[b]*nums[b]
b -= 1 #b从最右端向中间移
elif nums[a]*nums[a] > nums[b]*nums[b]: #则先将a放入新数组
ans[i] = nums[a]*nums[a]
a += 1 #从最左端向中间移
i -= 1 #i可往前移一位,且不管跳入哪个循环都能往前一位
return ans
双指针法:从正负交接往两边找
可理解为从中间向两边找,中间为正负组交界处。先通过enumerate遍历函数找到边界neg,在从neg出发向前找,neg+1出发向后找,比较出较小值后依次填充入新数组中
- enumerate(nums): 遍历函数,可返回数组nums的下标和对应数据
- 该方法要考虑边界问题,若负数组到边界,后续就是正数组往后加,反之亦然
- 且情况讨论时,需要先把边界情况定义好,否则可能会出现IndexError: list index out of range的报错
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
#先确定边界
neg = -1 ##若没有负数时
for i, num in enumerate(nums): #先确定边界
if num < 0:
neg = i
else:
break
n = len(nums)
a, b = neg, neg + 1
ans = list() #因是从前往后填充,可直接list.append
while a >= 0 or b < n: #只要满足其中一个条件,就可以接着填数;不满足某个条件是因为该组以达到边界
if a < 0: #负数组已达到边界,后续就是正数组往后加
ans.append(nums[b]*nums[b])
b += 1
elif b >= n: #正数组已达到边界,后续就是负数组往后加
ans.append(nums[a]*nums[a])
a -= 1
#需要先把边界情况定义好,否则可能会出现IndexError: list index out of range的报错
elif nums[a]*nums[a] <= nums[b]*nums[b]: #则先将a放入新数组
ans.append(nums[a]*nums[a])
a -= 1
elif nums[a]*nums[a] > nums[b]*nums[b]: #则先将b放入新数组
ans.append(nums[b]*nums[b])
b += 1
return ans
2. Q209. 长度最小的子数组
给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度最小的连续子数组,并返回其长度。如果不存在符合条件的子数组,返回 0。
暴力解法
用两个for循环嵌套,第一个设定起点i,第二个设定终点j,若满足和>target,则ans为满足条件的跨度区间最小值
class Solution:
def minSubArrayLen(self, target: int, nums: List[int]) -> int:
n = len(nums)
ans = n+1 #ans的最大值比较限制,也为若找不到满足条件和的判断标准
for i in range(n): #设定起点
total = 0
for j in range(i,n): #设定终点
total += nums[j]
if total >= target:
ans = min (ans, j-i+1) #筛选跨度区间最小值
break
return 0 if ans == n+1 else ans
滑动窗口
先固定滑动窗口终止位置b,以数组开头为滑动窗口起始位置a,若a到b和小于target则继续向后移动终止位置b,若大于target则后移a,一点点缩小窗口范围,看是否还符合条件
class Solution:
def minSubArrayLen(self, target: int, nums: List[int]) -> int:
n = len(nums)
ans = n+1
a = 0
total = 0
for b in range(n):
total += nums[b]
while total >=target: #若满足条件,则移动a缩小范围
ans = min(ans, b-a+1)
total -= nums[a] #剔除一位数字
a += 1
return 0 if ans == n+1 else ans
3. Q59. 螺旋矩阵Ⅱ
给一个正整数 n ,生成一个包含 1 到
n
2
n^2
n2所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
模拟法
画螺旋式可以上行从左到右,右列从上到下,下行从右到左,左列从下到上,一圈圈往内画,且最好用左闭右开区间
- 4种情况的讨论比较繁琐,需要注意i. j的增减
- range(start, end, n) 函数是左闭右开区间,不包括end, n 为每次移动步距,默认为1,若从大到小 range 需设定 n=-1
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
left, right, top, bottom = 0, n-1, 0, n-1 #划分边界,左右为列数,上下为行数
count, stop = 1, n*n #设定起始值和终点值
res = [[0] * n for _ in range(n)]
while count <= stop:
#从左向右填充最上列
for j in range(left,right+1):
res[top][j] = count
count +=1
top += 1 #上边界已填好一行
#从上向下填充最右列
for i in range(top,bottom+1):
res[i][right] = count
count +=1
right -= 1 #右边已填好一列
#从右向左填充最下列
for j in range(right,left-1,-1): #range为从大到小
res[bottom][j] = count
count +=1
bottom -= 1 #下边界已填好一行
#从下往上填充最左列
for i in range(bottom,top-1,-1):
res[i][left] = count
count +=1
left += 1 #右边已填好一列
return res