给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
思路:
和上一篇博客一个思路,也是转圈,不过打印成了赋值,知道n也就知道矩阵的行和列了。
int index = 1;
public int[][] generateMatrix(int n) {
int[][] arr = new int[n][n];
int a = 0, b = 0;
int c = arr.length - 1, d = arr[0].length - 1;
while (a <= c && b <= d && index <= n) { // 这个循环用来控制边框
helper(arr, a++, b++, c--, d--);
}
return arr;
}
private void helper(int[][] arr, int a, int b, int c, int d) {
// 判断同行的情况
if (a == c) {
for (int i = b; i <= d; i++) {
arr[a][i] = index++;
}
} else if (b == d) { // 判断同列的情况
for (int i = a; i <= c; i++) {
arr[i][b] = index++;
}
} else {
int curRow = a;
int curCol = b;
// 水平向右方向
while (curCol != d) {
arr[curRow][curCol++] = index++;
}
// 竖直向下方向
while (curRow != c) {
arr[curRow++][curCol] = index++;
}
// 水平向左方向
while (curCol != b) {
arr[curRow][curCol--] = index++;
}
// 竖直向上方向
while (curRow != a) {
arr[curRow--][curCol] = index++;
}
}
}