腾讯2016实习笔试：n阶蛇形矩阵

题目描述

输入一个整数n，从上到下、从左到右打印n阶蛇形矩阵。例如输入整数4，输出为：
1 　　2 　　3 　　4
12 　13 　 14 　 5
11 　16 　 15 　 6
10 　9 　　8 　　7

分析

生成一个n阶初始值为0的矩阵，从[0][0]位置开始依次向右、向下、向上、向左移动，每移动一次，填充一个自增1的数。移动方向变化的条件是达到矩阵的边界或者当前元素的下一个元素已经填充了数字（不为0）。生成了蛇形矩阵后，普通遍历打印输出即可。

代码：

public class SnakeMatrix {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        while(input.hasNext()) {
            int n = input.nextInt();
            printMatrix(n);
        }
        input.close();
    }

    public static void printMatrix(int n) {
        int[][] arr = new int[n][n];
        int x = 0, y = 0;
        int num = 1;  // num不能为0，否则死循环
        int sum = n * n;
        arr[0][0] = 1;
        while(num < sum) {
            // 向右移动，
            // 直到到达边界，或者向右移动的下一个元素不为0（新的已经填充的边界）
            while(y + 1 < n && arr[x][y + 1] == 0)
                arr[x][++y] = ++num;
            // 向下移动
            while(x + 1 < n && arr[x + 1][y] == 0)
                arr[++x][y] = ++num;
            // 向左移动
            while(y - 1 >= 0 && arr[x][y - 1] == 0)
                arr[x][--y] = ++num;
            // 向上移动
            while(x - 1 >= 0 && arr[x - 1][y] == 0)
                arr[--x][y] = ++num;
        }

        // 打印输出
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) {
                //System.out.println(arr[i][j] + "  ");
                // 带格式的打印 printf()
                System.out.printf("%4d", arr[i][j]);
            }
            System.out.println();
        }
    }
}