59. 螺旋矩阵 II

题目

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

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

题解

模拟

/**
 * @param {number} n
 * @return {number[][]}
 */
var generateMatrix = function(n) {
  let res = new Array(n);
  for (let i = 0; i < n; i++) {
    res[i] = new Array(n).fill(0);
  }
  let a = 0,
    b = n - 1;
  let num = 1;
  while (a < b) {
    // top
    for (let i = a; i <= b; i++) {
      res[a][i] = num++;
    }
    // right
    for (let i = a + 1; i <= b; i++) {
      res[i][b] = num++;
    }
    // bottom
    for (let i = b - 1; i >= a; i--) {
      res[b][i] = num++;
    }
    // left
    for (let i = b - 1; i > a; i--) {
      res[i][a] = num++;
    }
    a++;
    b--;
  }
  if (a == b) res[a][b] = num;
  return res;
};