螺旋矩阵II

螺旋矩阵II

难度：中等

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

示例 1：

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

示例 2：

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

提示：

• 1 <= n <= 20

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/spiral-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

class Solution {
    public int[][] generateMatrix(int n) {
        int top = 0, left = 0, right = n - 1, bottom = n - 1;
        int[][] ans = new int[n][n];
        int len = n * n;
        int x = 0, y = 0;
        int cur = 0;
        int[][] dirs = new int[][]{{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
        for (int i = 1; i <= len; i++) {
            ans[x][y] = i;
            if (cur == 0 && y == right) {
                cur += 1;
                top += 1;
            }else if (cur == 1 && x == bottom) {
                cur += 1;
                right -= 1;
            }else if (cur == 2 && y == left) {
                cur += 1;
                bottom -= 1;
            }else if (cur == 3 && x == top) {
                cur += 1;
                left += 1;
            }
            cur %= 4;
            x += dirs[cur][0];
            y += dirs[cur][1];
        }
        return ans;
    }
}