蛇形矩陣的代碼

蛇形矩陣是比較特別的矩陣，但是隻要掌握了座標的規律，用程序實現也不難。
以下是我用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之間!");
  
 }

}