螺旋矩阵 II

力扣题目:59. 螺旋矩阵 II

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

思路

看看我之前写的博客吧,不懂的欢迎评论问我,或者你直接代码也能看懂,挺简单的。
博客链接

AC代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        // 二维向量这么玩~
        // vector< vector > matrix(n, vector(n, 4)); // 二维vector,元素是向量,n个向量,每个向量是一个大小为n初值为4
        vector< vector<int> > matrix(n, vector<int>(n, 0)); // 二维vector,元素是向量,n个向量,每个向量是一个大小为n初值为0

        // 提前预判的思想:紫书的蛇形填数,提前预判 比 悔步 更好!
        int count = 0, x = 0, y = 0;    // 初始化坐标个填数个数count
        y = -1; // 因为提前预判,第一个数方便用
        while (count != n*n)    // 当等于n*n时,填完啦
        {
            // 向左,提前预判, 没越界 并且 未填入(则为初始值0)
            while (y + 1 < n && matrix[x][y + 1] == 0) matrix[x][ ++ y] = ++ count;
            // 向下,提前预判
            while (x + 1 < n && matrix[x + 1][y] == 0) matrix[ ++ x][y] = ++ count;
            // 向左
            while (y - 1 >= 0 && matrix[x][y - 1] == 0) matrix[x][ -- y] = ++ count;
            // 向上
            while (x - 1 >= 0 && matrix[x - 1][y] == 0) matrix[ -- x][y] = ++ count;
        }

        // for (int i = 0; i < n; i ++ )
        // {
        //     for (int j = 0; j < n; j ++ )
        //     {
        //         cout << matrix[i][j] << " ";
        //     }
        //     cout << endl;
        // }

        return matrix;
    }
};

你可能感兴趣的:(代码随想录,算法,c++)