【模拟】螺旋矩阵

一、题目

力扣原题：https://leetcode-cn.com/problems/spiral-matrix-ii/

二、模拟法

class Solution {
    public int[][] generateMatrix(int n) {
        // 初始化二维矩阵
        int[][] result = new int[n][n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                result[i][j] = -1;
            }
        }

        int max = n * n;
        int row = 0;
        int col = 0;
        // 右 = 1，下 = 2，左 = 3，上 = 4
        int direction = 1;
        for (int i = 1; i <= max; i++) {
            while (true) {
                // 未被占用
                if (result[row][col] == -1) {
                    result[row][col] = i;
                    break;
                } 

                // 往右
                if (direction == 1 && result[row][col + 1] == -1) {
                    col++;
                    if (col == n - 1 || result[row][col + 1] != -1) {
                        direction = 2;
                    }
                }

                // 往下
                else if (direction == 2 && result[row + 1][col] == -1) {
                    row++;
                    if (row == n - 1 || result[row + 1][col] != -1) {
                        direction = 3;
                    }
                }

                // 往左
                else if (direction == 3 && result[row][col - 1] == -1) {
                    col--;
                    if (col == 0 || result[row][col - 1] != -1) {
                        direction = 4;
                    }
                }

                // 往上
                else if (direction == 4 && result[row - 1][col] == -1) {
                    row--;
                    if (row == 0 || result[row - 1][col] != -1) {
                        direction = 1;
                    }
                }

                result[row][col] = i;
                break;
            }
        }
        return result;
    }
}

• 基本思路：没什么技巧，模拟就好了。从(0, 0)出发，按照右、下、左、上的优先次序填充数组。一旦选定了一个方向，需要沿着这条边走到底，代码中通过引入direction变量实现。另外，当走到边界或当前位置已被填充时，可以拐弯。
• 时间复杂度：O(n^2)。外层遍历1 - n^2，内层位置搜索的时间复杂度为O(1)。
• 空间复杂度：O(n^2)。需要二维数组记录放入的值。