解答一道面试题

看到网友发的一道面试题 试作解答

 

要求打印出：

int i = 5;

  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

int i = 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

思路很简单
把数据按照螺旋矩阵的路线装进数组
然后再打印出来

public class PrintSquare {
	private int[][] getArray(int n) {
		if (n > 0) {
			int[][] array = new int[n][n];
			int x = 0;
			int y = 0;
			int xStart = 0;
			int xEnd = n - 1;
			int yStart = 0;
			int yEnd = n - 1;
			for (int i = 1; i <= n * n; i++) {
				array[y][x] = i;
				if (y == yStart && x < xEnd) {// 横向前进
					x++;
					if (x == xEnd) {
						yStart++;
					}
				} else if (y < yEnd && x == xEnd) {// 纵向前进
					y++;
					if (y == yEnd) {
						xEnd--;
					}
				} else if (y == yEnd && x > xStart) {// 横向倒退
					x--;
					if (x == xStart) {
						yEnd--;
					}
				} else if (y > yStart && x == xStart) {// 纵向倒退
					y--;
					if (y == yStart) {
						xStart++;
					}
				}
			}
			return array;
		}
		return null;
	}

	public void printSquare(int n) {
		if (n > 0) {
			int[][] arr = getArray(n);
			for (int i = 0; i < n; i++) {
				for (int j = 0; j < n; j++) {
					System.out.print(arr[i][j] + "\t");
				}
				System.out.println();
			}
		}
	}

	public static void main(String[] args) {
		PrintSquare test = new PrintSquare();
		test.printSquare(5);
		System.out.println();
		test.printSquare(6);
	}

}