代码随想录算法训练营第二天| 977. 有序数组的平方、209. 长度最小的子数组、59. 螺旋矩阵

leetcode 977

leetcode题目链接
代码随想录文章讲解
代码随想录视频讲解

有序数组的平方

方法一：双指针法

重点

• 最大值一定从数组两边取
• 整体的取值应该是从两边逐渐往里收
• 使用左右指针把两边的值进行比对，再逐渐向里收
• 新开数组res = [-1] * len(nums)

完整python版答案

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        res = [-1] * len(nums)
        left = 0
        right = len(nums) - 1
        i = len(nums) - 1
        while left <= right:
            if nums[left] ** 2 > nums[right] ** 2:
                res[i] = nums[left] ** 2
                left += 1
            else:
                res[i] = nums[right] ** 2
                right -= 1
            i -= 1
        return res

方法二：暴力排序法

完整python版答案

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        for i in range(len(nums)):
            nums[i] *= nums[i]
        nums.sort()
        return nums

方法三：暴力排序法+列表推导法

完整python版答案

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        return sorted(x*x for x in nums)

leetcode 209

leetcode题目链接
代码随想录文章讲解
代码随想录视频讲解

长度最小的子数组

方法一：滑动窗口

重点

• 注意如果只有一个for循环，那么循环的应该是窗口的结束位置，而不是起始位置
• 当判断到 sum>target时，就要开始移动窗口的起始位置以找到长度最小的子数组了。这里起始位置的移动不是一次性的，而是sum>target条件下的多次移动。所以应该用while，而不是if
• 注意在每次移动起始位置时，sub应该等于j-i+1；另外，sum要减去上一个nums[i]（这两个点容易忘记）
• res的初始值设为float('inf')

完整python版答案

class Solution:
    def minSubArrayLen(self, target: int, nums: List[int]) -> int:
        i = 0
        sum = 0
        res = float('inf')
        for j in range(len(nums)):
            sum += nums[j]
            while sum >= target:
                sub = j - i + 1
                res = min(sub, res)
                sum -= nums[i]
                i += 1
        return res if res != float('inf') else 0

方法二：暴力法

完整python版答案

class Solution:
    def minSubArrayLen(self, s: int, nums: List[int]) -> int:
        l = len(nums)
        min_len = float('inf')
        
        for i in range(l):
            cur_sum = 0
            for j in range(i, l):
                cur_sum += nums[j]
                if cur_sum >= s:
                    min_len = min(min_len, j - i + 1)
                    break
        
        return min_len if min_len != float('inf') else 0

leetcode 59

leetcode题目链接
代码随想录文章讲解
代码随想录视频讲解

螺旋矩阵

方法：模拟过程

重点

• 每条边都是左闭右开
• 这道题算法的部分不多，有点不那么有趣

完整python版答案

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