螺旋矩阵算法练习

leetcode-59 螺旋矩阵Ⅱ

题目描述

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/spiral-matrix-ii

思路

看到这个题莫名想到递归，这个题无非就是一圈一圈来呗，从最外面一圈开始，先向右走到头，然后依次向下，向左，向上都走到头，接着开始里面一圈，直到全部遍历结束。

代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0));
        int x=0,y=0;
        int i=2;
        res[0][0]=1; //第一个元素放在循环里不太好处理，索性放外面了
        while(1){
            while(y+1<n&&res[x][y+1]==0){ //左边的条件是限制外圈的，右边的条件是限制内圈的
                res[x][++y]=i++;                
            }
            while(x+1<n&&res[x+1][y]==0){
                res[++x][y]=i++;                
            }
            while(y-1>=0&&res[x][y-1]==0){
                res[x][--y]=i++;                
            }
            while(x-1>=0&&res[x-1][y]==0){
                res[--x][y]=i++;                
            }
            
            if(i>n*n) break;
        }
        return res;
    }
};