蛇形矩阵的代码

蛇形矩阵是比较特别的矩阵，但是只要掌握了坐标的规律，用程序实现也不难。
以下是我用Java实现的蛇形矩阵：
n=4 的时候
输出为：
        1     2    6     7
        3     5    8     13
        4     9    12   14
       10    11   15   16

源代码实现如下：
       

import javax.swing.JOptionPane;
public class Test
{
 public static void main(String [] args)
 {
  String str;
  final int MAX=10;
  int input;
  int d,i,j,m;
  int A[][]=new int[10][10];
  str=JOptionPane.showInputDialog("输入:");
  input=Integer.parseInt(str);
  if(input>0&&input<MAX)
  {
   i=1;
   j=1;
   m=1;
   d=1;
   do
   {
    A[j]=m;
    switch(d)
    {
       case 1: j++;if(i==1)d=2;else d=4;break;
       case 2:   i++;j--;if(i==input)d=1;
     else if(j==1)d=3;break;
       case 3: i++;if(j==input)d=2;else d=4;break;

       case 4: j++;i--;if(j==input)d=3;else if(i==1)d=1;break;
  
    }
   m++;
   }while(m<=input*input);
   System.out.println("Output:");
   System.out.println();
   for(i=1;i<=input;i++)
   {
    for(j=1;j<=input;j++)
     {System.out.print(" "+A[j]);}
    System.out.println();
   }  
  }
  else
  System.out.println("输入的数不合法,必须在1~10之间!");
  
 }

}