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

69 阅读2分钟

977.有序数组的平方

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.长度最小的子数组

209.长度最小的子数组

文章讲解 视频讲解

重点:需要理解窗口的起始位置及结束位置如何移动。当sum >= target说明起始位置需要移动,通过减值来获取该窗口下的最小长度。结束位置是遍历数组的指针。

另外如何理解两层while的时间复杂度为O(n),在移动结束位置指针时,起始位置指针是不会重置的,只能向后移动,没有从头遍历。每个元素仅被操作两次,所以整体时间复杂度为2 x nO(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