JavaScript算法系列--leetcode螺旋矩阵

给定一个包含 m x n 个元素的矩阵(m 行, n 列),请按照顺时针螺旋顺序,返回矩阵中的所有元素。

示例 1:

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

示例 2:

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

解法如下:

/**
 * @param {number[][]} matrix
 * @return {number[]}
 */
var spiralOrder = function(matrix) {
    if (!matrix.length) return []
    if (matrix.length == 1) return matrix[0]
    // 向左前进:'left',向右前进:'right',向上前进:'top', 向下前进:'down'
    let direction = 'right'
    let x = 0, y = 0
    const visitedArr = []
    const res = []
    const col = matrix[0].length
    const row = matrix.length
    for (let i = 0; i < row; i++) {
      visitedArr[i] = []
      for (let j = 0; j < col; j++) {
        visitedArr[i][j] = false
      }
    }
    for (let index = 0; index < row * col; index++) {
      switch (direction) {
        case 'right':
          for (let k = x; k < col; k++) {
            if (!visitedArr[y][k]) {
              res.push(matrix[y][k])
              visitedArr[y][k] = true
              x = k
              if (x === col - 1) direction = 'down'
            } else {
              direction = 'down'
            }
          }
          break;
        case 'down':
          for (let k = y + 1; k < row; k++) {
            if (!visitedArr[k][x]) {
              res.push(matrix[k][x])
              visitedArr[k][x] = true
              y = k
              if (y === row - 1) direction = 'left'
            } else {
              direction = 'left'
            }
          }
          break;
        case 'left':
          for (let k = x - 1; k >= 0; k--) {
            if (!visitedArr[y][k]) {
              res.push(matrix[y][k])
              visitedArr[y][k] = true
              x = k
              if (x === 0) direction = 'top'
            } else {
              direction = 'top'
            }
          }
          break;
        case 'top':
          for (let k = y - 1; k >= 0; k--) {
            if (!visitedArr[k][x]) {
              res.push(matrix[k][x])
              visitedArr[k][x] = true
              y = k
            } else {
              direction = 'right'
            }
          }
          break;

        default:
          break;
      }
    }
    return res
};

如有更好解法,欢迎留言探讨。

你可能感兴趣的:(Javascript算法系列,前端)