解密 LeetCode 2326：巧用 Python 旋转二维数组，征服 Spiral Matrix IV

2023-10-05 16:18:58

旋转二维数组，巧解 LeetCode 2326：Spiral Matrix IV

LeetCode 2326：Spiral Matrix IV 是一道难度为 Medium 的算法题，旨在考察选手对二维数组旋转操作的掌握程度。本题要求将一个二维数组按照螺旋顺序旋转指定的次数，并返回旋转后的结果。

算法原理：构建辅助数组，逐层旋转

LeetCode 2326：Spiral Matrix IV 的解题思路是，首先构建一个辅助数组，将二维数组中的元素按照螺旋顺序排列在辅助数组中，然后将辅助数组中的元素复制回原二维数组中。具体步骤如下：

初始化辅助数组和循环变量。
确定螺旋顺序的起始位置和结束位置。
按照螺旋顺序将元素从二维数组复制到辅助数组中。
将辅助数组中的元素复制回原二维数组中。
重复步骤 2-4，直到旋转指定的次数。

Python 代码实现：简洁高效

def spiralMatrixIV(A, k):
  """
  :type A: List[List[int]]
  :type k: int
  :rtype: List[List[int]]
  """
  m, n = len(A), len(A[0])
  辅助数组 = [[0] * n for _ in range(m)]

  # 旋转 k 次
  for _ in range(k):
    # 初始化循环变量
    top, bottom, left, right = 0, m - 1, 0, n - 1
    # 确定螺旋顺序的起始位置和结束位置
    while left <= right and top <= bottom:
      # 从左到右
      for i in range(left, right + 1):
        辅助数组[top][i] = A[top][i]
      # 从上到下
      for i in range(top + 1, bottom):
        辅助数组[i][right] = A[i][right]
      # 从右到左
      if top < bottom:
        for i in range(right, left, -1):
          辅助数组[bottom][i] = A[bottom][i]
      # 从下到上
      if left < right:
        for i in range(bottom - 1, top, -1):
          辅助数组[i][left] = A[i][left]

      # 更新循环变量
      top += 1
      bottom -= 1
      left += 1
      right -= 1

    # 将辅助数组中的元素复制回原二维数组中
    A = 辅助数组

  return A