从零学算法59

59 .螺旋矩阵 II

给你一个正整数 n ，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

示例 1：

输入：n = 3

输出：[[1,2,3],[8,9,4],[7,6,5]]

示例 2：

输入：n = 1

输出：[[1]]

提示：

1 <= n <= 20

• 按顺序定义顺时针的四个方向 [[0, 1], [1, 0], [0, -1], [-1, 0]]，初始从 (0,-1) 开始，如果走一步后发现出界或遇到已填充的格子就回退一步并更新方向，否则就填值

  var generateMatrix = function (n) {
      const pos = [[0, 1], [1, 0], [0, -1], [-1, 0]]
      let posIdx = 0
      let i = 0, j = -1
      let genCount = 0
      const ans = new Array(n).fill(0).map(() => new Array(n).fill(0))
      for (let val = 1; val < n * n; val++) {
          // 走一步
          i += pos[posIdx][0]
          j += pos[posIdx][1]
          // 如果出界或遇到已填充的格子就回退一步并更新方向
          if (i < 0 || i >= n || j < 0 || j >= n || ans[i][j] != 0) {
              i -= pos[posIdx][0]
              j -= pos[posIdx][1]
              posIdx = (posIdx + 1) % 4
              continue
          }
          ans[i][j] = val
      }
      return ans
  };