螺旋矩阵 II

[louzhedong]

习题

出处：LeetCode 算法第59题

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

示例:

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

思路

同螺旋矩阵

解答

/**
 * @param {number} n
 * @return {number[][]}
 */
var generateMatrix = function (n) {
  var result = [];
  for (var i = 0; i < n; i++) {
    result[i] = [];
  }
  var row = n - 1;
  var col = n - 1;
  var r = 0;
  var c = 0;
  var current = 1;
  while (r <= row && c <= col) {
    for (var i = r; i <= col; i++) {
      result[r][i] = current++;
    }
    r++;
    for (var i = r; i <= row; i++) {
      result[i][col] = current++;
    }
    col--;
    if (r <= row) {
      for (var i = col; i >= c; i--) {
        result[row][i] = current++;
      }
      row--;
    }
    if (c <= col) {
      for (var i = row; i >= r; i--) {
        result[i][c] = current++;
      }
      c++;
    }
  }

  return result;
};

console.log(generateMatrix(3));