给你一个正整数 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
思路:
这道题是leetcode54-螺旋矩阵-CSDN博客的一个变体,这个思路还可以按照螺旋数组进行遍历,只不过在遍历的时候不用把数组加入列表中了,而是应该填入数据,之后我们返回添加后的二维数组就可以了。
//IDEA测试代码
public static void main(String[] args) {
//leetcode中重写了toString方法,我们用这个简单的测试一下就可以
int[][] h = generateMatrix(3);
for (int i = 0; i < h.length; i++) {
for (int j = 0; j < h[i].length; j++) {
System.out.print(h[i][j]);
}
System.out.println();
}
}
public static int[][] generateMatrix(int n) {
int[][]matrix = new int[n][n];
int size = 1;
//用来收缩第一行和最后一行
int firstRow = 0, lastRow = n-1;
//用来收缩第一列和最后一列
int firstCol = 0, lastCol = n-1;
//进入循环
while (size <= n*n) {
//→
for (int i = firstCol; i <= lastCol; i++) matrix[firstRow][i] = size++;
//↓
if (firstRow != lastRow) {
for (int i = firstRow + 1; i <= lastRow; i++) matrix[i][lastCol] = size++;
//←
if (firstRow != lastRow && firstCol != lastCol) {
for (int i = lastCol - 1; i >= firstCol; i--) matrix[lastRow][i] = size++;
//↑
if ((firstRow + 1) != lastRow)
for (int i = lastRow - 1; i > firstRow; i--) matrix[i][firstCol] = size++;
}
}
//进行收缩
firstRow++;
lastRow--;
firstCol++;
lastCol--;
}
return matrix;
}
总结:
假如不做螺旋矩阵,直接做螺旋矩阵II的话,是比较难想到循环遍历这个方法的。leetcode通过引导1和2,使得2这个题还是比较好解决的。