题目描述
给你一个正整数 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 {
int[][] ans;
public int[][] generateMatrix(int n) {
ans = new int[n][n];
right(0, 0, 1, n); // 先向右赋值
return ans;
}
public void right(int i, int j, int count, int n) { // 行,列,数字,n
ans[i][j] = count; // 先填充当前数字
if(j != n-1 && ans[i][j+1] == 0) { // 如果没到右边界且右边元素未赋值过数字,则可以继续往右赋值
right(i, j+1, count+1, n);
} else if((j == n-1 || ans[i][j+1] != 0) && (i != n-1 && ans[i+1][j] == 0)) {
// 如果达到右边界或者右边元素已赋值过数字,且未达到下边界且下边元素未赋值过数字,则可以继续向下赋值
down(i+1, j, count+1, n);
}
}
// 向下,同向右赋值
public void down(int i, int j, int count, int n) {
ans[i][j] = count;
if(i != n-1 && ans[i+1][j] == 0) {
down(i+1, j, count+1, n);
} else if((i == n-1 || ans[i+1][j] != 0) && (j != 0 && ans[i][j-1] == 0)) {
left(i, j-1, count+1, n);
}
}
// 向左,同向右赋值
public void left(int i, int j, int count, int n) {
ans[i][j] = count;
if(j != 0 && ans[i][j-1] == 0) {
left(i, j-1, count+1, n);
} else if((j == 0 || ans[i][j-1] != 0) && (i != 0 && ans[i-1][j] == 0)) {
up(i-1, j, count+1, n);
}
}
// 向上,同向右赋值
public void up(int i, int j, int count, int n) {
ans[i][j] = count;
if(i != 0 && ans[i-1][j] == 0) {
up(i-1, j, count+1, n);
} else if((i == 0 || ans[i-1][j] != 0) && (j != n-1 && ans[i][j+1] == 0)) {
right(i, j+1, count+1, n);
}
}
}
注意点
- 虽然代码长了点,不过整体看下来还是很清晰的
- 友情链接:【LeetCode】第54题——螺旋矩阵(难度:中等),正好操作是反过来的