【leetcode】54. 螺旋矩阵

目录

• 题目
• 模拟

题目

模拟

没做出来...看的灵神的题解。

原题解：https://leetcode.cn/problems/spiral-matrix/solutions/2966229/liang-chong-fang-fa-jian-ji-gao-xiao-pyt-4wzk/?envType=problem-list-v2&envId=iZ0YYZP1

自己想的时候一直纠结于怎么变m和n的值来遍历，虽然这也是一种方法，不过看到方向向量的解法真是又学到了。

用方向向量记录右下左上四个方向，遍历过的值进行标记，如果越界或已经访问过，就向右转90°。遍历总个数是m*n。

/**
 * @param {number[][]} matrix
 * @return {number[]}
 */
var spiralOrder = function(matrix) {
    const direction = [[0, 1], [1, 0], [0, -1], [-1, 0]]; // 方向向量，右下左上
    const m = matrix.length, n = matrix[0].length;
    const res = new Array(m * n);
    let i = 0, j = 0, di = 0;
    for (let k = 0; k < m * n; k++) {
        res[k] = matrix[i][j] // 加入结果
        matrix[i][j] = Infinity; // 标记
        // 下一步的位置
        const x = i + direction[di][0];
        const y = j + direction[di][1];
        if (x < 0 || x >= m || y < 0 || y >= n || matrix[x][y] === Infinity) {
            di = (di + 1) % 4;
        } 
        i += direction[di][0];
        j += direction[di][1];
    }
    return res;
};