蛇形矩阵

本题的来源是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;}