力扣59. 螺旋矩阵 II（按层模拟）

力扣59. 螺旋矩阵 II（按层模拟）

https://leetcode-cn.com/problems/spiral-matrix-ii/

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

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

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

按层模拟

复杂度分析

时间复杂度：O(n^2)，其中 n 是给定的正整数。矩阵的大小是 n×n，需要填入矩阵中的每个元素。

空间复杂度：O(1)。除了返回的矩阵以外，空间复杂度是常数。

// 59generateMatrix.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

#include <iostream>
#include<vector>
using namespace std;
class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>>result (n, vector<int>(n));
        int top = 0;
        int down = n - 1;
        int left = 0;
        int right = n - 1;
        int num = 1;
        while (1)
        {
            //从左往右，直到最右，top下移一行，如果top>down，break
            for (int i = left; i <= right; i++)
            {
                result[top][i] = num;
                num++;
            }
            top++;
            if (top>down)
            {
                break;
            }
            //从上往下，直到最下，right左移一列，如果right<left，break
            for (int i = top; i <= down; i++)
            {
                result[i][right] = num;
                num++;
            }
            right--;
            if (right < left)
            {
                break;
            }
            //从右往左，直到最左，down上移一行，如果down<top，break
            for (int i = right; i >= left; i--)
            {
                result[down][i] = num;
                num++;
            }
            down--;
            if (down < top)
            {
                break;
            }
            //从下往上，直到最上，left右移一列，如果left>right，break
            for (int i = down; i >= top; i--)
            {
                result[i][left] = num;
                num++;
            }
            left++;
            if (left > right)
            {
                break;
            }
        }
        return result;
    }
};
int main()
{
    Solution s;
    auto result = s.generateMatrix(3);
    for (int i = 0; i < result.size(); i++)
    {
        for (int j = 0; j < result[0].size(); j++)
        {
            cout << result[i][j] << " ";
        }
        cout << endl;
    }
    std::cout << "Hello World!\n";
}