【LeetCode59.螺旋矩阵II】——二维数组遍历

最新推荐文章于 2024-07-25 09:47:49 发布

木瓜星灵TT

最新推荐文章于 2024-07-25 09:47:49 发布

阅读量298

点赞数 3

分类专栏： LeetCode学习笔记文章标签：矩阵算法 leetcode 学习 c++

本文链接：https://blog.csdn.net/weixin_51630192/article/details/126440640

版权

LeetCode学习笔记专栏收录该内容

37 篇文章 0 订阅

订阅专栏

59.螺旋矩阵II

题目链接

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

示例 1：

输入：n = 3
输出：[[1,2,3],[8,9,4],[7,6,5]]

示例 2：

输入：n = 1
输出：[[1]]

提示：

1 <= n <= 20

思路：

本题总体而言没有什么特定的算法，算是对于二维数组的一次运用，关键点就是要不重不漏地进行遍历赋值。

代码：

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0)); // 使用vector定义一个二维数组
        int startx = 0, starty = 0; // 定义每循环一个圈的起始位置
        int loop = n / 2; // 每个圈循环的次数
        int mid = n / 2; // 矩阵中间的位置，用于处理n为奇数时的情况
        int count = 1; // 用来给矩阵中每一个空格赋值，从1开始
        int offset = 1; //每次循环缩小边界
        int i, j;//用于进行循环
        while (loop--) { //进行loop次循环
            i = startx;
            j = starty;

            // 填充上行从左到右(左闭右开)
            for (j = starty; j < n - offset; j++) {
                res[startx][j] = count++;
            }
            // 填充右列从上到下(左闭右开)
            for (i = startx; i < n - offset; i++) {
                res[i][j] = count++;
            }
            // 填充下行从右到左(左闭右开)
            for (; j > starty; j--) {
                res[i][j] = count++;
            }
            // 模拟填充左列从下到上(左闭右开)
            for (; i > startx; i--) {
                res[i][j] = count++;
            }

            // 第二圈开始的时候，起始位置要各自加1
            startx++;
            starty++;

            // offset 控制每一圈里每一条边遍历的长度
            offset += 1;
        }

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