[Leetcode]59.螺旋矩陣Ⅱ

給定一個正整數 n，生成一個包含 1 到 n2 所有元素，且元素按順時針順序螺旋排列的正方形矩陣。

示例:

輸入: 3
輸出:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

Solution:

蛇形環繞，為了減少判斷或者迴圈的程式碼，我們需要環繞一圈不變的變數作為參照量。

於是！我們設定一個變數i，這個i的意思表示第i外層。

一圈的填數如下:

從左到右，從i行i列->i行n-i-1列

從上到下，從i+1行n-i-1列->n-i-1行n-i-1列

從右到左，從n-i-1行n-i-2列->n-i-1行i列

從下到上，從n-i-2行i列->i+1行i列

同時我們設定一個變數count，用於計數保證迴圈結束。

class Solution {
public:
  vector<vector<int>> generateMatrix(int n) { 
      vector<vector<int>> ans(n,vector<int>(n));
      int count = 1,i=0;
      while(count<=n*n){
        int j = i;
        while(j<n-i)
          ans[i][j++] = count++;
        j = i+1;
        while(j<n-i)
          ans[j++][n - i-1] = count++;
        j = n - i-2;
        while(j>i)
          ans[n -i-1][j--] = count++;
        j = n-i-1;
        while(j>=i+1)
          ans[j--][i] = count++;
        i++;
    }
    return ans;
};