59. 螺旋矩阵 II

59. 螺旋矩阵 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

题解

思路

这是一道分析逻辑题目,没有涉及到具体算法知识,但是要弄清楚他的螺旋逻辑

• n阶矩阵最外层有4*(n-1)个元素
  • 也就是每一圈有四种情况, 每种情况需要(n-1)次
• 一共有n/2圈(解法2)
• 最外圈执行完了之后, 第二圈可以看作是一个n-2阶的矩阵, 循环上个步骤

解法1是基于四种填充情况来写(→ ↓ ← ↑)

解法2是基于每一圈来写,共有n/2圈, 单独处理n为奇数的情况

代码

class Solution {
    public int[][] generateMatrix(int n) {
        int[][] matrix = new int[n][n];
        int x, y;
        x = 0; y = 0;
        int flag = 0; // 0,1,2,3
        int count = 0;
        int matrixEdge = n;
        for(int i = 1; i <= n*n; i++) {
            if (flag == 0) {
                matrix[x][y++] = i;
                if (++count == matrixEdge - 1) {
                    flag = 1;
                    count = 0;
                }
            } else if (flag == 1) {
                matrix[x++][y] = i;
                if (++count == matrixEdge - 1) {
                    flag = 2;
                    count = 0;
                }
            } else if (flag == 2) {
                matrix[x][y--] = i;
                if (++count == matrixEdge - 1) {
                    flag = 3;
                    count = 0;
                }
            } else if (flag == 3) {
                matrix[x--][y] = i;
                if (++count == matrixEdge - 1) {
                    flag = 0;
                    count = 0;
                    x++;
                    y++;
                    matrixEdge -= 2;
                }
            }
        }

        return matrix;
    }



    public int[][] solution2(int n) {
        int loop = 0;  // 控制循环次数
        int[][] res = new int[n][n];
        int start = 0;  // 每次循环的开始点(start, start)
        int count = 1;  // 定义填充数字
        int i, j;

        /**
            一共需要转n/2圈
         */
        while (loop++ < n / 2) { // 判断边界后，loop从1开始
            // 模拟上侧从左到右
            for (j = start; j < n - loop; j++) {
                res[start][j] = count++;
            }

            // 模拟右侧从上到下
            for (i = start; i < n - loop; i++) {
                res[i][j] = count++;
            }

            // 模拟下侧从右到左
            for (; j >= loop; j--) {
                res[i][j] = count++;
            }

            // 模拟左侧从下到上
            for (; i >= loop; i--) {
                res[i][j] = count++;
            }
            start++;
        }

        if (n % 2 == 1) {
            res[start][start] = count;
        }

        return res;
    }
}