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