java - java中的螺旋矩阵

Question

我正在尝试根据用户输入创建一个矩阵。我想开始如下所示

输入:3

输出:创建如下所示的 3x3 矩阵，从 1 开始，然后是 2，然后添加前两个数字。

3   2    1
5   55   34  
8   13   21

如果用户输入为 4，那么它应该创建一个 4x4 矩阵。

Answer 1

试试这个。

static int[][] DIRS = {{0, -1}, {1, 0}, {0, 1}, {-1, 0}};

static boolean canGo(int[][] a, int n, int r, int c) {
    if (r < 0 || r >= n)
        return false;
    if (c < 0 || c >= n)
        return false;
    if (a[r][c] > 0)
        return false;
    return true;
}

static int[][] solve(int n) {
    int max = n * n;
    int[][] array = new int[n][n];
    int dir = 0, prev = 1, cur = 1;
    int row = 0, col = n - 1;
    for (int i = 1; i <= max; ++i) {
        array[row][col] = cur;
        if (!canGo(array, n, row + DIRS[dir][0], col + DIRS[dir][1]))
            dir = (dir + 1) % DIRS.length;
        row += DIRS[dir][0];
        col += DIRS[dir][1];
        int temp = cur;
        cur = prev + cur;
        prev = temp;
    }
    return array;
}

和

int[][] a = solve(3);
for (int[] row : a)
    System.out.println(Arrays.toString(row));

结果:

[3, 2, 1]
[5, 55, 34]
[8, 13, 21]