螺旋矩阵II

题目信息

给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

示例1:


matrix3_3.jpg

输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]

解题思路

  1. 暴力破解:
  2. 无效操作分析:
  3. 优化方法:
  4. 考虑边界
  5. 编码实现

代码

class Solution {
    public int[][] generateMatrix(int n) {
        if (n <= 0) {
            return new int[1][1];
        }
        int maxNum = n * n;
        int[][] matrix = new int[n][n];
        int row = 0, column = 0;
        // 定义方向矩阵
        int[][] directions = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}}; // 右下左上
        int directionIndex = 0;
        for (int curNum = 1; curNum <= maxNum; curNum++) {
            matrix[row][column] = curNum;
            // 计算下一个横纵坐标
            int nextRow = row + directions[directionIndex][0], nextColumn = column + directions[directionIndex][1];
            // 判断坐标的合法性和当前位置是否已经赋值
            if (nextRow < 0 || nextRow >= n || nextColumn < 0 || nextColumn >= n || matrix[nextRow][nextColumn] != 0) {
                directionIndex = (directionIndex + 1) % 4; // 顺时针旋转至下一个方向
            }
            row = row + directions[directionIndex][0];
            column = column + directions[directionIndex][1];
        }
        return matrix;
    }
}

题目来源:力扣(LeetCode)
题目链接:https://leetcode-cn.com/problems/spiral-matrix-ii

商业转载请联系官方授权,非商业转载请注明出处。

你可能感兴趣的:(螺旋矩阵II)