【Leetcode019】螺旋矩阵 II

59、螺旋矩阵 II【中等】

给你一个正整数 n ，生成一个包含 1 到 n² 所有元素，且元素按顺时针顺序螺旋排列的 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(N²)
• 空间复杂度：O(N²)

方法二：按层遍历

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(N²)
• 空间复杂度：O(N²)