螺旋矩阵的斜向填入法

相信很多人做螺旋矩阵这道题都会真的模拟螺旋的方式输出。我尝试了一种新的方式：斜向填入。

其具体方法概述如下：

拿5阶举例：首先填充基准边，即1、2、3、4、5、6、7、8、9。通过基准边可以推出其对称位的数字。如16、15、14、13、12、11、10。那么5阶就可以降为4阶（5阶不过就是比4阶多了两条边嘛）。将新得到的这些数字作为基准边，重复操作即可不断降阶，直到填出所有的数字。那该如何推导呢？怎么得到增量？我们不妨列出每一路完全推演。如从5开始的一次推演：

5->13->19->23->25。我们从1开始一直列出从1到9开始的完全推演，可以得到下面这张图。

不难发现，这些增量是有规律的。横向代表某起点的一路完全推演。纵向是某一路完全推演的第n次推演的特征。偶数次推演都是从2开始增，成公差为2的等差数列。奇数次推演都是从某个数字开始，公差为-2递减。并且你会发现，从第2次推演开始，都是基于第一次推演的这个数列的。只需要在这个数列上加上偏移量，就可以得到后面推演的子数列。

具体代码如下：

/**
 * Project:HelixPhalanx
 * Author:Zhang Cuiyan
 * Date:2024/3/10
 *
 * 螺旋矩阵的斜向填入法
 */
public class HelixPhalanx {
    private static final int n = 4;

    public static void main(String[] args) {
        HelixPhalanx.print(HelixPhalanx.northWest(n));
    }

    public static void print(int[][] mat) {
        if (mat == null) return;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                System.out.print(mat[i][j] + "\t");
            }
            System.out.println();
        }
    }

    public static int[][] northWest(int n) {
        final int[][] mat = new int[n][n];
        //从1,1开始
//        int[][] result = new int[n+2][n+2];
        final int[] addArray = new int[2 * n - 1];
        //得到增值数组
        // 如5阶为0 14 12 10 8 6 4 2 0
        for (int i = 1; i < 2 * n - 2; i++) {
            addArray[2 * n - 2 - i] = 2 * i;
        }
        //填入基准边
        for (int i = 0; i < n; i++) {
            mat[0][i] += i + 1;
        }
        for (int i = 1; i < n; i++) {
            mat[i][n - 1] += n + i;
        }
        /**
         * 斜向填入数字
         * 遍历n阶基准边
         * 偶数阶，奇数阶分别斜向对称
         * i为路 j为维度 第i路总共j个边
         * n阶总共n个边
         * 奇数nPos = j - 1 偶数nPos = j - 2
         * 实际nPos是 0 2 4 .....。
         * 奇数边从左开始偏移，偶数边从右开始偏移
         */
        for (int i = 0; i < 2 * n - 1; i++) {
            //奇偶阶交替
            int nums = i;//对应路的边数
            int x = 0;
            int y = i;
            if (i >= n) {
                x = i % n + 1;
                y = n - 1;
                nums = 2 * n - 2 - i;
            }
            for (int j = 1; j <= nums; j++) {
                int oldNum = mat[x][y];
                if (j % 2 == 1) {
                    //更新坐标
                    int temp = x;
                    x = y;
                    y = temp;
                    mat[x][y] += addArray[i + j - 1] + oldNum;
                } else {
                    int temp = y;
                    y = x - 1;
                    x = temp + 1;
                    //addArray.length - (i - j + 2)
                    //2*n-1-i+j-2
                    mat[x][y] += addArray[2 * n - 3 - i + j] + oldNum;
                }
            }
        }
        return mat;
    }
}