生成n*n蛇形矩阵的算法

来源：http://www.blogjava.net/nokiaguy/archive/2009/07/24/288163.html

在描述算法之前，先看看下面的5*5的表格：

1	3	4	10	11
2	5	9	12	19
6	8	13	18	20
7	14	17	21	24
15	16	22	23	25

    上面的表格很容易看出规律。就是从左上角第一个格开始（起始为1），然后延右上角到左下角的斜线。先从下到上，再从上到下。开始按数字递增排列。也就是说每一个斜线上分别有如下几组数字：
1    2 3     4 5 6       7 8 9 10      11 12 13 14 15          16 17 18 19      20 21 22      23 24       25
    由于是先从上到下（1可以看做是从上到下），再从下到上，很象一条蛇，因此，该数字表格也可称为蛇形矩阵。现在要与一个方法（或函数），方法的参数是一个int类型，表示n，方法返回一个二维数组，表示要获得的往返接力数字表格。     实际上，这个算法并不复杂，只需要从分别获得1至n^2中每个数字对应的二维数组的坐标就可以了。先拿这个5行5列的表格来说，求出上面每组数组对应的坐标（起始位置为0）。

                                
    从上面的从标可以看出一个规律。  左上角的半个表格（以对角线分界）的横坐标和纵坐标从0开始，每一组增1，直到增至表格的边界（n - 1），而且是交替的，也就是说，偶数行是列增，行减小，行+列=组的索引。而右下角的4组数字虽然行、列也是交替增长的，但递减的行或列总是从(n - 1)开始（对于本例，是从4开始），而递增的行或列总是从index - n + 1开始，其中index表示组的索引。这就可以得出一个算法。实现代码如下：

public static int[][] getGrid(int n)
{
    int[][] array = new int[n][n];
    int row = 0, col = 0, m = 1;
    //  用于控制奇偶组，false表示偶组，true表示奇组
    boolean isRow = false;
    //  i表示当前组的索引，从0开始
    for (int i = 0; i < (2 * n - 1); i++)
    {
        row = i;
        while (row >= ((i < n) ? 0 : i - n + 1))
        {
            //  如果处理的是右下角表格中的数字，行或列最大不能超过n-1
            if (row > (n - 1))
                row = n - 1;
            col = i - row;
            if (isRow)
                array[row][col] = m;
            else  //  将row变成列，将col变成行
                array[col][row] = m;
            m++;
            row--;
        }
        //  切换奇偶组
        isRow = !isRow;
    }
    return array;
}

  另一种算法

  
上面实现的算法需要循环N*N次才可以生成蛇形矩阵。但仔细分析一下，还可以稍微变换一下这个算法，使循环次数减小至N*N/2。我们上学时曾学过用高斯的方法计算1+2+3+...+100，
  1 + 100 = 101，2 + 99 = 101，...，50+51 = 101，因此，结果是101 * 50 =
5050。很方便。我们这个算法也可采用类似的方法。仔细观察上面5*5的数字表格发现，算出左上角的矩阵中每一个数字后，都可以直接获得右下角度某个位置的数字。例如在(0,0)位置的1，可以向到(4,4)位置的25，(1,2)位置的9可以得到(3,2)位置的17。我们发现，每一对数之和都为26。而且它们坐标的关系是(row,col)，(n
- row - 1, n - col -
1)。因此，只要得到左上角的半个矩阵，就可以得出右下角的另外半个矩阵。如果n为奇数，对角线中间的一个数（在5*5的矩阵中是13）与之对应的数是其自身。好，我们看看改进的算法的实现：

public static int[][] getGrid1(int n)
{
    int[][] array = new int[n][n];
    int row = 0, col = 0, m = 1;
    int number1 =  (n * n / 2 + n * n % 2);
    int number2 = n * n + 1;        
    boolean isRow = false;
    //  number1表示要计算的蛇形矩阵中最大的数字，对于5*5矩阵来说该数是13
    for (int i = 0; m < number1; i++)
    {
        row = i;
        while (row >= 0)
        {
            col = i - row;
            if (isRow)
            {
                array[row][col] = m;
                //  填充与m对应的另外一个数
                array[n - row - 1][n - col - 1] = number2 - m;
            }
            else
            {
                array[col][row] = m;
                //  填充与m对应的另外一个数
                array[n - col - 1][n - row - 1] = number2 - m;

            }
            m++;
            if(m >= number1) break;
            row--;
        }
        isRow = !isRow;
    }
    return array;
}

 上面的算法虽然将循环次数减少了一半，但每次循环的计算量增加了，因此，算法总体效率并没有提高。至于使用哪个算法，可根据实际情况决定。    如果想输出n=10的数字表格，可以使用int[][] grid = getGrid(10)或int[][] grid1 = getGrid1(10)，会得到同样的结果。输出grid和grid1，看看是不是下面的结果：

1	3	4	10	11	21	22	36	37	55
2	5	9	12	20	23	35	38	54	56
6	8	13	19	24	34	39	53	57	72
7	14	18	25	33	40	52	58	71	73
15	17	26	32	41	51	59	70	74	85
16	27	31	42	50	60	69	75	84	86
28	30	43	49	61	68	76	83	87	94
29	44	48	62	67	77	82	88	93	95
45	47	63	66	78	81	89	92	96	99
46	64	65	79	80	90	91	97	98	100