leetcode#59 螺旋矩阵

leetcode#59 螺旋矩阵

题目：

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

示例：

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

思路：

不难看出每次都经历右下左上的顺序，每遍历一排或者一列，相应的边界就会减一，比如遍历了123，那么向上走的边界就会减一，以此类推，所以我们可以设置边界l,r,u,d，每次遍历后使边界减一就可以了。不难看出遍历顺序为从左边界到右边界，从上边界到下边界，等等。

代码：

class Solution
{
public:
    vector<vector<int>> generateMatrix(int n)
    {
        vector<vector<int>> ans(n, vector<int>(n));
        int l = 0, r = n-1, u = 0, d = n-1;
        int cnt = 0;
        while (cnt < n * n)
        {
            for (int i = l; i <= r;++i)
                ans[u][i] = ++cnt;
            ++u;
            for (int i = u; i <= d;++i)
                ans[i][r] = ++cnt;
            --r;
            for (int i = r; i >= l;--i)
                ans[d][i] = ++cnt;
            --d;
            for (int i = d; i >= u;--i)
                ans[i][l] = ++cnt;
            ++l;
        }
        return ans;
    }
};