Java 實現順時針螺旋二維陣列輸出

看到一道有點意思的邏輯演算法題，便著手實現一下。列印 N*N順時針螺旋陣列，規律如下：

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

Java 實現

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {

        Scanner in = new Scanner(System.in);
        int n = in.nextInt();

        int[][] rect = new int[n + 1][n + 1];

        int x = 0, y = 1; // 當前遍歷的位置
        int px = 1, py = 0; // x/y 在當前迴圈方向的增量
        int pn = n; // 該方向上還需要多少個數字（會遞減，因為順時針往裡縮）
        int sum = n; // 當前遍歷方向的終點位置

        /** 目標輸出順時針 */
        // 1       2       3       4       5
        // 16      17      18      19      6
        // 15      24      25      20      7
        // 14      23      22      21      8
        // 13      12      11      10      9

        /** 翻轉矩陣為逆時針（為了便於後面邏輯處理，輸出的時候再翻轉回去） */
        // 1       16      15      14      13
        // 2       17      24      23      12
        // 3       18      25      22      11
        // 4       19      20      21      10
        // 5       6       7       8       9
        for (int i = 1; i <= n * n; i++) { // 翻轉矩陣迴圈
            // 移到下一個點(第一次方向1->5 x增量為1， y的增量為0， 第二次方向為6->9,x增量為0，y增量為1，pn減少一個)
            x += px;
            y += py;
            rect[x][y] = i;
            if (i == sum) { // 到達該方向的最後一個點
                if (px == 1 || px == -1) { // x的增量不為0，則為豎向 -> 橫向
                    pn = pn - 1; // 豎向變為橫向的時候，該方向上數字的個數會減少一個
                    sum = sum + pn;
                    py = px;
                    px = 0;
                } else { // x的增量為0， y的增量不為0，則為橫向 -> 豎向
                    sum = sum + pn;
                    px = -py;
                    py = 0;
                }
            }
        }

        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= n; j++) {
                System.out.print(rect[j][i] + "\t"); // 翻轉復原輸出
            }
            System.out.println();
        }
    }

}

輸出測試