给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
难度:中等。
算法思路:其实在做完力扣54题,这题就很简单了,之前我在学习算法的时候也写过这道题,不过当时是采用递归完成的,这次用DFS去完成它,其实稍微在54题的代码上更改一下就可以了。
class Solution {
public int[][] generateMatrix(int n) {
int[][] nums = new int[n][n];
boolean[][] visited = new boolean[n][n];
int total = n*n;
int x = 0, y = 0;
int[][] directions = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
int directionIndex = 0;
for (int i = 0; i < total; i++) {
nums[x][y] = i+1;
visited[x][y] = true;//经过这个位置了,将其标记
//下一个位置的坐标
int next_x = x + directions[directionIndex][0], next_y = y + directions[directionIndex][1];
//判断下一个位置是否不满足条件,如果不满足,则将改变方向移动
if (next_x < 0 || next_x >= n || next_y < 0 || next_y >= n || visited[next_x][next_y]) {
directionIndex = (directionIndex + 1) % 4;
}
x += directions[directionIndex][0];
y += directions[directionIndex][1];
}
return nums;
}
}
执行用时:
0 ms ,在所有 Java 提交中击败了100%的用户
内存消耗:
36.4 MB, 在所有 Java 提交中击败了75.11%的用户