59、螺旋矩阵 II【中等】
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
方法一:模拟
1.1 思路分析
这个跟螺旋矩阵思路一样,只不过那个是读数据,这个是写数据,根据题可知,最后形成的一定是n*n的矩形。
1.2 代码实现
func generateMatrix(n int) [][]int {
ans := make([][]int, n)
visited := make([][]int, n)
for i:=0; i<n; i++{
ans[i] = make([]int, n)
visited[i] = make([]int, n)
}
var (
row, col = 0, 0
directions = [][]int{[]int{0,1}, []int{1,0}, []int{0,-1}, []int{-1,0}}
directionIndex = 0
)
for i:=1; i<=n*n;i++{
ans[row][col] = i
visited[row][col] = 1
nextRow, nextCol := row+directions[directionIndex][0], col+directions[directionIndex][1]
if nextCol<0 || nextCol>=n || nextRow<0 || nextRow>=n || visited[nextRow][nextCol]==1{
directionIndex = (directionIndex+1) % 4
}
row += directions[directionIndex][0]
col += directions[directionIndex][1]
}
return ans
}
1.3 测试结果
1.4 复杂度
- 时间复杂度:O(N2)
- 空间复杂度:O(N2)
方法二:按层遍历
2.1 思路分析
这个跟螺旋矩阵思路一样,只不过那个是读数据,这个是写数据,根据题可知,最后形成的一定是n*n的矩形。
2.2 代码实现
- go
func generateMatrix(n int) [][]int {
ans := make([][]int, n) // 结果矩阵,n*n
for i:=0; i<n; i++{
ans[i] = make([]int, n)
}
top, left, right, bottom := 0, 0, n-1, n-1
total := n*n // total主要是判断是否结束
for k:=1; k<=n*n&&total>0; {
for i:=left; i<=right; i++{ // 最上层从左向右
ans[top][i] = k
total--
k++
}
top++
for i:=top; i<=bottom; i++{ // 最右层从上向下
ans[i][right] = k
total--
k++
}
right--
for i:=right; i>=left; i--{ // 最下层从右向左
ans[bottom][i] = k
total--
k++
}
bottom--
for i:=bottom; i>=top; i--{ // 最左层从下向上
ans[i][left] = k
total--
k++
}
left++
}
return ans
}
- java
class Solution {
public int[][] generateMatrix(int n) {
int left = 0, right = n-1;
int top = 0, bottom = n-1;
int[][] res = new int[n][n];
int k = 1;
while(k <= n*n){
for(int i=left; i<=right; ++i){
res[top][i] = k;
k++;
}
top++;
for(int i=top; i<=bottom; ++i){
res[i][right] = k;
k++;
}
right--;
for(int i=right; i>=left; --i){
res[bottom][i] = k;
k++;
}
bottom--;
for(int i=bottom; i>=top; --i){
res[i][left] = k;
k++;
}
left++;
}
return res;
}
}
2.3 测试结果
- go
- java
2.4 复杂度
- 时间复杂度:O(N2)
- 空间复杂度:O(N2)