很久前遇到过的一个面试题,让写一个螺旋矩阵,如下图:
一串数字先向右、再向下、再向左、再向上循环,直到填满整个矩阵。
我不太喜欢去算行列每个位置和n的关系,我把它看成贪吃蛇。
一条蛇从左上角开始移动,遇到墙壁只能拐弯;
遇到自己已经走过的位置,拐弯。
这样的思路就够了,然后用n去规定一下墙壁,用二维数组的值规定一下走过的位置,一步步走下去,就可以实现。
直接上代码了,我对代码进行了一个简单的方法抽取,看着方便点儿。
public static void main(String[] args) {
test(10);
}
public static void test(int n) {
// 规定一个方向且有序,右(x + 1, y)right,下(x, y + 1)down,左(x - 1, y)left,上(x, y - 1)up
int x = 1; // 列 1 - n, 规定一个范围,1 <= x && x <= n, 1 <= y && y <= n
int y = 1; // 行 1 - n
String d = "right";// 默认往右
int[][] arr = new int[n][n];//初始化一个二维数组
int value = 1;// 每移动一步 + 1
arr[0][0] = value;// 初始位置的初始值
// 因为初始值已经有了,所以这里需要走 n*n - 1 步
for (int i = 1; i < n * n; i++) {
value ++;
// 判断 1-有没有越界 2-前面的值有没有数
if(needTurn(arr, n, d, x,y)) {
d = turn(d);//拐弯
}
//移动
if ("right".equals(d)) {
x = x + 1;
} else if ("down".equals(d)) {
y = y + 1;
} else if ("left".equals(d)) {
x = x - 1;
} else if ("up".equals(d)) {
y = y - 1;
}
//赋值
arr[y - 1][x - 1] = value;
print(arr, n);//这里的打印可以监控每一步的数组变化
}
print(arr, n);// 打印
}
public static boolean needTurn(int[][] arr, int n, String d, int x, int y) {
if ("right".equals(d)) {
if (x == n || arr[y - 1][x] > 0) {
return true;
}
} else if ("down".equals(d)) {
if (y == n || arr[y][x - 1] > 0) {
return true;
}
} else if ("left".equals(d)) {
if (x == 1 || arr[y - 1][x - 2] > 0) {
return true;
}
} else if ("up".equals(d)) {
if (y == 1 || arr[y - 2][x - 1] > 0) {
return true;
}
}
return false;
}
public static void print(int[][] arr, int n) {
for(int i = 0 ; i < n ; i ++) {
for(int j = 0 ; j < n ; j ++) {
System.out.print(arr[i][j] < 10 ? " " + arr[i][j] : " " + arr[i][j]);
}
System.out.println();
}
}
public static String turn(String d) {
if("right".equals(d)) {
return "down";
}
if("down".equals(d)) {
return "left";
}
if("left".equals(d)) {
return "up";
}
if("up".equals(d)) {
return "right";
}
return "";
}