天天看点

【精选高频面试题】Leetcode.59.螺旋数组Ⅱ

Leetcode.59.螺旋数组Ⅱ

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

给你一个正整数

n

,生成一个包含

1

n2

所有元素,且元素按顺时针顺序螺旋排列的

n x n

正方形矩阵

matrix

【精选高频面试题】Leetcode.59.螺旋数组Ⅱ
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
           

示例 2:

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

本题思路:本题的关键在于如何得到一个n*n的螺旋数组。其本质就是一个二维数组。然后螺旋式的给这个二维数组赋值,在赋值过程中,要让你的循环条件始终不变,我们采用的是左闭右开的原则,比如上面的数组,我们先添加1,2,再3,4,再5,6 … 。同时考虑根据n的大小设置中间值的大小

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>>res(n,vector<int>(n,0));//使用vector创建二维数组
        int startx=0,starty=0;//初始化坐标
        int loop=n/2;//循环圈数;
        int mid=n/2;//可能得到的中间值
        int count=1;//赋值的初值
        int offset=1;//控制赋值的范围值
        int i,j;
        while(loop--){
           i=startx;
           j=starty;
           
           for(;j<starty+n-offset;j++){
               res[startx][j]=count++;
           }
           for(;i<startx+n-offset;i++){
               res[i][j]=count++;
           }
           for(;j>starty;j--){
               res[i][j]=count++;
           }
           for(;i>startx;i--){
               res[i][j]=count++;
           }
           startx++;
           starty++; //更新到内层的初始元素,准备进行内层的螺旋化。
           offset+=2; //因为前后的元素都有值
        }
        if(n%2){
                res[mid][mid]=count;
            }
            return res;
        }
        
    
};