蛇形矩阵

本题的来源是2016年的腾讯春季校招笔试题。

我只是被人抱了大腿。(o´・ェ・｀o)

试题描述

蛇形矩阵：作为一种常用的数学数列，是由1开始的自然数依次排列成的N*N的正方形矩阵，数字依次由外向内递增，如下面实例：
n = 3的蛇形矩阵
1 2 3
8 9 4
7 6 5

n = 6的蛇形矩阵
1 2 3 4 5 6
20 21 22 23 24 7
19 32 33 34 25 8
18 31 36 35 26 9
17 30 29 28 27 10
16 15 14 13 12 11

此题要求输入蛇形矩阵的宽度N，输出整个蛇形矩阵结果，注意输出格式要求按照矩阵从上到下的依次按行输出。每行中间无需换行输出。

样例输入：3
样例输出：1 2 3 8 9 4 7 6 5

解题思路

纯模拟题。

只需要考虑“撞”到边界后的处理就好。撞到边界后同时需要更新边界的值，因为边界会越来越小。

源代码

import java.util.Scanner;

public class Solution {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);

        int n = in.nextInt();
        int[][] matrix = new int[n][n];

        int arrowX = 1, arrowY = 0;
        int borderLeftX = 0, borderRightX = n - 1;
        int borderTopY = 1, borderBottomY = n - 1;
        int currentX = 0, currentY = 0;

        for ( int i = 1; i <= n * n; ++ i ) {
            matrix[currentY][currentX] = i;

            currentX += arrowX;
            currentY += arrowY;

            if ( arrowX == 1 && arrowY == 0 && currentX == borderRightX ) {
                // 此时正在向右走，并且到达了右侧边界
                -- borderRightX;

                arrowX = 0;
                arrowY = 1;
            } else if ( arrowX == 0 && arrowY == 1 && currentY == borderBottomY ) {
                // 此时正在向下走，到达了下侧边界
                -- borderBottomY;

                arrowX = -1;
                arrowY = 0;
            } else if ( arrowX == -1 && arrowY == 0 && currentX == borderLeftX ) {
                // 此时正在向左走，到达了左侧边界
                ++ borderLeftX;

                arrowX = 0;
                arrowY = -1;
            } else if ( arrowX == 0 && arrowY == -1 && currentY == borderTopY ) {
                // 此时正在向上走，到达了上部边界
                ++ borderTopY;

                arrowX = 1;
                arrowY = 0;
            }
        }

        for ( int i = 0; i < n; ++ i ) {
            for ( int j = 0; j < n; ++ j ) {
                System.out.print(matrix[i][j] + " ");
            }
        }
    }

    private static final int N = 100;
}