59. 螺旋矩阵 II | LeetCode算法题解

题目描述

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

示例:

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

思路

创建一个 n*n 的数组，依次把 1 到 n²放入数组。
同时控制好右,下,左,上四个方向，每次到达边界或下一个位置已经有元素了，即调整下一个方向。
数字放完，即得到结果。

解法

/**
 * @param {number} n
 * @return {number[][]}
 * */
const generateMatrix = n => {
  const result = Array(n)
    .fill()
    .map(() => []);

  let direction = 'right';
  let i = 0;
  let j = 0;
  for (let num = 1; num <= n * n; num++) {
    result[i][j] = num;

    switch (direction) {
      case 'right':
        if (j + 1 >= n || result[i][j + 1]) {
          direction = 'down';
          i++;
        } else {
          j++;
        }
        break;
      case 'down':
        if (i + 1 >= n || result[i + 1][j]) {
          direction = 'left';
          j--;
        } else {
          i++;
        }
        break;
      case 'left':
        if (j - 1 < 0 || result[i][j - 1]) {
          direction = 'up';
          i--;
        } else {
          j--;
        }
        break;
      case 'up':
        if (i - 1 < 0 || result[i - 1][j]) {
          direction = 'right';
          j++;
        } else {
          i--;
        }
        break;
      default:
        break;
    }
  }

  return result;
};