59. Spiral Matrix II

題目：

Given a positive integer n, generate an n x n matrix filled with elements from 1 to n2 in spiral order.

 

Example 1:

Input: n = 3
Output: [[1,2,3],[8,9,4],[7,6,5]]

Example 2:

Input: n = 1
Output: [[1]]

 

Constraints:

• 1 <= n <= 20

 

思路：

題意是按照順時針一層一層列印出一個n*n的矩陣。思路就是分清楚四個邊界，然後每列印一層，就更新相對應層的值。如原本up是0，down是n-1，列印完第一行，up就應該變1；列印完最後一行，down就應該變n-2。最後注意迴圈跳出情況，當up>down時跳出，而不是up>=down，因為偶數時，up在第二層，down在第三層時，down的那層是還沒列印的。

 

程式碼：

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