977.有序数组的平方
想法:可用双指针,一个从前往后,一个从后往前,两个指针做对比,大的放到新数组的最末尾。如果本题是无序数组的话不能这么做。
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
i = 0
j = len(nums) - 1
k = j
res = [0] * (j + 1)
while i <= j:
if nums[i] * nums[i] > nums[j] * nums[j]:
res[k] = nums[i] * nums[i]
i += 1
else:
res[k] = nums[j] * nums[j]
j -= 1
k -= 1
return res
209.长度最小的子数组
重点:需要理解窗口的起始位置及结束位置如何移动。当sum >= target说明起始位置需要移动,通过减值来获取该窗口下的最小长度。结束位置是遍历数组的指针。
另外如何理解两层while的时间复杂度为O(n),在移动结束位置指针时,起始位置指针是不会重置的,只能向后移动,没有从头遍历。每个元素仅被操作两次,所以整体时间复杂度为2 x n即O(n)。
同时result最小长度初始化不能为0。
class Solution:
def minSubArrayLen(self, target: int, nums: List[int]) -> int:
size = len(nums) - 1
i,j = 0, 0
result = float('inf')
sum = 0
while j <= size:
sum += nums[j]
while sum >= target:
result = min(result, j - i + 1)
sum = sum - nums[i]
i += 1
j += 1
return result if result != float('inf') else 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