代码随想录训练营第二天|977|209|59
977 有序数组的平方
核心:数组其实是有序的, 只不过负数平方之后可能成为最大数了。
那么数组平方的最大值就在数组的两端,不是最左边就是最右边,不可能是中间。
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
left, right, i = 0, len(nums)-1, len(nums)-1
res = [float("inf")]*len(nums) # 设置成无穷大的方法
while i >= 0:
if nums[right]**2 > nums[left]**2:
res[i] = nums[right]**2
right -= 1
else:
res[i] = nums[left]**2
left += 1
i -= 1
return res
209 长度最小的子数组
双指针
class Solution:
def minSubArrayLen(self, s: int, nums: List[int]) -> int:
lenght = len(nums)
right, left = 0, 0
min_len = float("inf")
cur_sum = 0
while right < lenght:
cur_sum += nums[right]
while cur_sum >= s:
min_len = min(min_len, (right - left+1))
cur_sum -= nums[left]
left += 1
right += 1
return min_len if min_len != float("inf") else 0 # 不等于无穷时候即为min_len,等于记为0
59.螺旋矩阵II
主要是找好循环不变量,左闭右开,省着乱。
设置横纵坐标时候用手画一下记录一下更方便准确。
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
nums = [[0] * n for _ in range(n)]
startx, starty = 0, 0 # 起始点
loop, mid = n // 2, n // 2 # 迭代次数、n为奇数时,矩阵的中心点
count = 1 # 计数
for offset in range(1, loop + 1) : # 每循环一层偏移量加1,偏移量从1开始
for i in range(starty, n - offset) : # 从左至右,左闭右开
nums[startx][i] = count
count += 1
for i in range(startx, n - offset) : # 从上至下
nums[i][n - offset] = count
count += 1
for i in range(n - offset, starty, -1) : # 从右至左
nums[n - offset][i] = count
count += 1
for i in range(n - offset, startx, -1) : # 从下至上
nums[i][starty] = count
count += 1
startx += 1 # 更新起始点
starty += 1
if n % 2 != 0 : # n为奇数时,填充中心点
nums[mid][mid] = count
return nums
查看19道真题和解析