【精選高頻面試題】Leetcode.59.螺旋數組Ⅱ

Leetcode.59.螺旋數組Ⅱ

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

給你一個正整數

n

，生成一個包含

1

到

n2

所有元素，且元素按順時針順序螺旋排列的

n x n

正方形矩陣

matrix

。

輸入：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;
        }
        
    
};