螺旋矩阵 II
59. 螺旋矩阵 II
给你一个正整数 n
,生成一个包含 1
到 n^2
所有元素,且元素按顺时针顺序螺旋排列的 n x n
正方形矩阵 matrix
。
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
提示:
1 <= n <= 20
巧遍历-四种遍历方向,分类处理
类似54. 螺旋矩阵。
class Solution {
public int[][] generateMatrix(int n) {
int l = 0, r = n - 1, t = 0, b = n - 1;
int[][] mat = new int[n][n];
int num = 1, tar = n * n;
while (num <= tar) {
// 注意顺序一定不能错,即一定要连续填充。
for (int i = l; i <= r; i++) mat[t][i] = num++; // 从左到右填充上边界
t++;
for (int i = t; i <= b; i++) mat[i][r] = num++; // 从上到下填充右边界
r--;
for (int i = r; i >= l; i--) mat[b][i] = num++; // 从右到左填充下边界
b--;
for (int i = b; i >= t; i--) mat[i][l] = num++; // 从下到上填充左边界
l++;
}
return mat;
}
}
- 时间复杂度:O(n^2),其中 n 是给定的正整数。矩阵的大小是 n×n,需要填入矩阵中的每个元素。
- 空间复杂度:O(1)。除了返回的矩阵以外,空间复杂度是常数。