leetcode 59. 螺旋矩阵 II

​
​

作者简介：C/C++ 、Golang 领域耕耘者，创作者
个人主页：作者主页
活动地址：CSDN21天学习挑战赛
题目来源： leetcode官网
如果感觉博主的文章还不错的话，还请关注➕ 、点赞 、收藏三连支持一下博主哦~~~

题目描述

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

示例1：

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

示例2：

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

算法分析

此题方法是模拟填数就行。

可以看下之前的leetcode54 螺旋矩阵

我们顺时针定义四个方向：上右下左。
从左上角（0， 0）开始遍历，先往右走，走到不能走为止，然后更改到下个方向，再走到不能走为止，依次类推，遍历 n² 个格子后停止。

代码实现

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> re(n, vector<int>(n));

        // 答案数组直接作为遍历的标记， 初始化默认为0
        int dx[] = {0, 1, 0 , -1}, dy[] = {1, 0, -1, 0};
        for(int i = 1, x = 0, y = 0, d = 0; i <= n * n; i ++)
        {
            re[x][y] = i;
            int a = x + dx[d], b = y + dy[d];
            if(a < 0 || a >=n || b < 0 || b >= n || re[a][b])
            {
                d = (d + 1) % 4;
                a = x + dx[d], b = y + dy[d];
            }
            x = a, y = b;
        }

        return re;
    }
};

执行结果：

时间复杂度分析

矩阵中每个格子遍历一遍，时间复杂度为O(n²)

如果觉得对你有帮助的话:
点赞，你的认可是我创作的动力！
收藏，你的青睐是我努力的方向！
✏️ 评论，你的意见是我进步的财富！