Leetcode第59题(螺旋矩阵2)

题目描述:

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

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

示例 2：
输入：n = 1
输出：[[1]]

此题与54题类似, 就不再过多赘述了。

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        int up = 0,down = n - 1,left = 0,right = n -1;
        vector<vector<int>> ans(n,vector<int>(n));
        int num = 0;
        while(num <= n * n){
            for(int i = left; i <= right; i++) ans[up][i] = ++num;
            if(++up > down) break;

            for(int i = up; i <= down; i++) ans[i][right] = ++num;
            if(--right < left) break;

            for(int i = right; i >= left; i--) ans[down][i] = ++num;
            if(--down < up) break;

            for(int i = down; i >= up; i--) ans[i][left] = ++num;
            if(++left > right) break;   
        }
        return ans;
    }
};