算法(6)-螺旋矩阵 II

数组

 2024/03/06 

题目(medium)：

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

思路

坚持循环不变量原则。

坚持了每条边左闭右开的原则。

const generateMatrix = (n) => {
    let startX = startY = 0;   // 起始位置
    let loop = Math.floor(n/2);   // 旋转圈数
    let mid = Math.floor(n/2);    // 中间位置
    let offset = 1;    // 控制每一层填充元素个数
    let count = 1;     // 更新填充数字
    let res = new Array(n).fill(0).map(() => new Array(n).fill(0));

    while (loop--) {
        let row = startX, col = startY;
        // 上行从左到右（左闭右开）
        for (; col < n - offset; col++) {
            res[row][col] = count++;
        }
        // 右列从上到下（左闭右开）
        for (; row < n - offset; row++) {
            res[row][col] = count++;
        }
        // 下行从右到左（左闭右开）
        for (; col > startY; col--) {
            res[row][col] = count++;
        }
        // 左列做下到上（左闭右开）
        for (; row > startX; row--) {
            res[row][col] = count++;
        }

        // 更新起始位置
        startX++;
        startY++;

        // 更新offset
        offset += 1;
    }
    // 如果n为奇数的话，需要单独给矩阵最中间的位置赋值
    if (n % 2 === 1) {
        res[mid][mid] = count;
    }
    return res;
}

原文作者：Yizhou Shen

原文链接：http://shenyizhou.github.io/2024/03/06/2024-03-06-%E7%AE%97%E6%B3%95(6)-%E8%9E%BA%E6%97%8B%E7%9F%A9%E9%98%B5%20II/

发表日期：March 6th 2024, 11:22:32 pm

更新日期：March 17th 2024, 3:24:40 pm

Next Post

算法(7)-链表
Previous Post

算法(5)-长度最小的子数组和滑动窗口

CATALOG

1. 题目(medium)：
2. 思路