面试题——蛇形矩阵

AcWing 756 蛇形矩阵

题目

输入两个整数n和m，输出一个n行m列的矩阵，将数字1到n * m按照回字蛇形填充至矩阵中。

具体矩阵形式可参考样例。

输入格式

输入共一行，包含两个整数n和m。

输出格式

输出满足要求的矩阵。

矩阵占n行，每行包含m个空格隔开的整数。

数据范围

1≤n,m≤1001≤n,m≤100

输入样例：

3 3

输出样例：

1 2 3
8 9 4
7 6 5

思路

​ 这道题是要将大小为1 ~ n * m的数字按照顺时针绕圈（有点白话）的方式放入到一个n * m的数组中，所以做法十分明显，我们需要定义_右下左上四个方向的偏移量_，通过在现有坐标上加上方向偏移量来实现我们当前指针（也就是二维数组下标）的移动

​ 现在怎么移动的问题解决了，另一个问题就是判断我们_下一次要移动的位置是否已经超出矩阵边界_，或者_下一次要移动的位置是否已经被赋过值_了，通过这两个条件，我们可以判断我们的坐标移动是不是需要_拐弯_了，如果是，只需要让偏移量改变即可，另外由于矩阵中数字的范围是1 ~ n * m，所以凡是下一个位置检测到!= 0的数字，那就说明这个位置已经被走过了

​ 下面我们一起来看代码

代码

#include<iostream>

using namespace std;

int res[110][110];

int main() {
	int n, m;
	int i = 0, j = -1;
	int item = 0;
	int dx[4] = { 0, 1, 0, -1 };
	int dy[4] = { 1, 0, -1, 0 };
	scanf("%d%d", &n, &m);
	int x = 1;
	while (x <= n * m) {
		if (i + dx[item] >= n || j + dy[item] >= m || j + dy[item] == -1 || res[i + dx[item]][j + dy[item]]) {
			item++;
		}
		else {
			i += dx[item];
			j += dy[item];
			res[i][j] = x;
			x++;
		}
		if (item == 4)
			item = 0;
	}
	for (i = 0; i < n; i++) {
		for (j = 0; j < m; j++)
			printf("%d ", res[i][j]);
		puts("");
	}
}