题目描述
给你一个正整数 n ,生成一个包含 1 到 n^2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例1
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例2
输入:n = 1
输出:[[1]]
思路
参考LeetCode Krahets 大神 的思路(人家的方法咋就这么简单高效呢)
JAVA代码
class Solution {
public int[][] generateMatrix(int n) {
int l = 0, r = n - 1, t = 0, b = n - 1;
int num = 1, target = n * n;
int[][] mat = new int[n][n];
while (num <= target) {
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(n2),我们遍历一次矩阵,矩阵元素个数为*(n2),所以时间复杂度为O*(n^2)
空间复杂度:O(n^2),需要一个二维矩阵存储结果