[leetcode]59. 螺旋矩阵 II

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

示例:

输入: 3
输出:

[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/spiral-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

方法:
设置low,hight,left,right四个边界量,由于最终矩阵为一个正方形,所以每次的螺旋不用判断low行和left列是否存在。直接循环每个螺旋的就可以了。
注意vector的初始化:vector<vector<int>> res(n, vector<int>(n, 0))

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        
        vector<vector<int>> res(n, vector<int>(n, 0));
        
        int high = 0, low = n - 1;
        int left = 0, right = n - 1;
        int num = 0;
        
        while (low >= high && left <= right) {
            for (int i = left; i <= right; i++)
                res[high][i] = ++num;
            for (int i = high + 1; i <= low; i++)
                res[i][right] = ++num;
            for (int i = right - 1; i >= left; i--)
                res[low][i] = ++num;
            for (int i = low - 1; i > high; i--)
                res[i][left] = ++num;

            high++;
            right--;
            low--;
            left++;
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/prettysky123/article/details/91893748