蛇形矩阵
题目来源:微软面试题
时间限制:1000ms 内存限制:64mb
题目描述
输入两个整数n和m,输出一个
n
n
n行
m
m
m列的矩阵,将数字
1
1
1 到
n
∗
m
n*m
n∗m 按照回字蛇形填充至矩阵中。
具体矩阵形式可参考样例。
输入格式
输入共一行,包含两个整数 n n n和 m m m。
输出格式
输出满足要求的矩阵。
矩阵占
n
n
n行,每行包含
m
m
m个空格隔开的整数。
数据范围
1 ≤ n , m ≤ 100 1 ≤ n,m ≤ 100 1≤n,m≤100
样例输入
3 3
样例输出
1 2 3
8 9 4
7 6 5
解题思路
分析需要判断的格子,为上、下、左、右,四个格子。
上、下、左、右,四个方向也是填数的方向。
将上、右、下、左四个方向分别用0,1,2,3表示,即:上=0,右=1,下=2,左=3。
所以先用一个二维数组来表示
x
x
x和
y
y
y两个坐标的变化值,即:int[][] vec = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}};
然后遍历所有格子即:遍历n*m个格子
将数填入格子内,如果判定为撞墙了,则方向+1,如此进行下去。
具体的判定方法在代码的注释里,如果有不明白的地方,欢迎在评论里告诉我。
解题代码-Java
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner input = new Scanner(System.in);
int n = input.nextInt();
int m = input.nextInt();
input.close();
int[][] vec = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}}; //表示4个方向x和y的变化值
int[][] res = new int[n][m]; //保存结果的数组
int x = 0, y = 0, d = 1; //定义初始坐标和方向
for (int i = 1; i <= n * m; i++) { //遍历所有格子
res[x][y] = i; //将数填入格子
int a = x + vec[d][0], b = y + vec[d][1]; //获取下一个格子的坐标
if (a >= n || a < 0 || b >= m || b < 0 || res[a][b] != 0) {
//判断是否撞墙,以下是判定为撞墙的所有情况
//1.x坐标不在 [0-n] 区间内
//2.y坐标不在 [0-m] 区间内
//3.方向不变的下一个格子已经被填入了数字
//如果判定为撞墙,则改变方向,并更新下一个格子的坐标为更改方向后的下一个格子
d = (d + 1) % 4;
a = x + vec[d][0];
b = y + vec[d][1];
}
//将坐标改为下一个格子的坐标
x = a;
y = b;
}
for (int i = 0; i < n; i++) { //循环输出二维数组
for (int j = 0; j < m; j++) {
System.out.print(res[i][j] + " ");
}
System.out.println();
}
}
}