代码随想录算法训练营第二天 | 977.有序数组的平方、209.长度最小的子数组、 59.螺旋矩阵II
LeetCode 977-有序数组的平方
题目:
给你一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
示例 1: 输入:nums = [-4,-1,0,3,10] 输出:[0,1,9,16,100] 解释:平方后,数组变为 [16,1,0,9,100],排序后,数组变为 [0,1,9,16,100]
示例 2: 输入:nums = [-7,-3,2,3,11] 输出:[4,9,9,49,121]
题目链接:977-有序数组的平方
思路:双指针,一前一后比较,选择乘积最大的数字放在数组最后
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
n = len(nums)
ans = [0] * n
i, j, pos = 0, n-1, n-1
while i <= j: #大的数字放数组最后
if nums[i] * nums[i] > nums[j] * nums[j]:
ans[pos] = nums[i] * nums[i]
i += 1
else:
ans[pos] = nums[j] * nums[j]
j -= 1
pos -= 1
return ans
LeetCode 209-长度最小的子数组
题目:
给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度最小的 连续 子数组,并返回其长度。如果不存在符合条件的子数组,返回 0。
示例:
输入:s = 7, nums = [2,3,1,2,4,3] 输出:2 解释:子数组 [4,3] 是该条件下的长度最小的子数组。
思路:双指针,右指针确定窗口的中止位置,先移动右指针找到符合题目要求的子数组再缩小左右指针的范围。
题目链接:209-长度最小的子数组
class Solution:
def minSubArrayLen(self, target: int, nums: List[int]) -> int:
i, j = 0, 0
res = float("inf")
Sum = 0
for j in range(len(nums)):
Sum += nums[j]
while Sum >= target:
res = min(res, j - i + 1) #缩小左右指针范围观察是否仍然符合条件要求
Sum -= nums[i]
i += 1 #移动起始指针
return 0 if res == float("inf") else res
LeetCode 59-螺旋矩阵 II
题目:
给定一个正整数 n,生成一个包含 1 到 n^2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3 输出: [ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]
思路: 循环遍历,每次遍历到边界的倒数第二个数停下,遍历完一圈后缩小范围继续遍历。
题目链接:59-螺旋矩阵 II
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
res = [[0]*n for i in range(n)]
cnt = 1
left, right, up, down = 0, n-1, 0, n-1
while cnt < n * n: # 若是<= 会导致数组多循环一次
for i in range(left, right):
res[up][i] = cnt
cnt += 1
for i in range(up, down):
res[i][right] = cnt
cnt += 1
for i in range(right, left, -1):
res[down][i] = cnt
cnt += 1
for i in range(down, up, -1):
res[i][left] = cnt
cnt += 1
up += 1 # 缩小范围
right -= 1
down -= 1
left += 1
if n % 2 == 1: # n为奇数时,填充中心点
res[n // 2][n // 2] = cnt
return res
总结: 今天的题目不算太难,理解起来仍算容易,但是本人在螺旋矩阵这道题花了比较多时间确定矩阵的边界,在数组第一个while循环条件中一开始错误使用了"<="导致数组多做了一次循环。在经过今天的训练,个人对数组的边界理解更加深刻。