leetcode59-螺旋矩阵II

给你一个正整数 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

思路：

        这道题是leetcode54-螺旋矩阵-CSDN博客的一个变体，这个思路还可以按照螺旋数组进行遍历，只不过在遍历的时候不用把数组加入列表中了，而是应该填入数据，之后我们返回添加后的二维数组就可以了。

 //IDEA测试代码
    public static void main(String[] args) {
        //leetcode中重写了toString方法，我们用这个简单的测试一下就可以
        int[][] h = generateMatrix(3);
        for (int i = 0; i < h.length; i++) {
            for (int j = 0; j < h[i].length; j++) {
                System.out.print(h[i][j]);
            }
            System.out.println();
        }

    }

public  static  int[][] generateMatrix(int n) {
        int[][]matrix = new int[n][n];
        int size = 1;

        //用来收缩第一行和最后一行
        int firstRow = 0, lastRow = n-1;
        //用来收缩第一列和最后一列
        int firstCol = 0, lastCol = n-1;

        //进入循环
        while (size <= n*n) {
            //→
            for (int i = firstCol; i <= lastCol; i++) matrix[firstRow][i] = size++;
            //↓
            if (firstRow != lastRow) {
                for (int i = firstRow + 1; i <= lastRow; i++) matrix[i][lastCol] = size++;
                //←
                if (firstRow != lastRow && firstCol != lastCol) {
                    for (int i = lastCol - 1; i >= firstCol; i--) matrix[lastRow][i] = size++;
                    //↑
                    if ((firstRow + 1) != lastRow)
                        for (int i = lastRow - 1; i > firstRow; i--) matrix[i][firstCol] = size++;
                }
            }

            //进行收缩
            firstRow++;
            lastRow--;
            firstCol++;
            lastCol--;
        }

        return matrix;
    }

总结：

        假如不做螺旋矩阵，直接做螺旋矩阵II的话，是比较难想到循环遍历这个方法的。leetcode通过引导1和2，使得2这个题还是比较好解决的。