蛇形矩阵
#算法笔试面试辅导
1.题目描述
输入两个整数 nn 和 mm,输出一个 nn 行 mm 列的矩阵,将数字 11 到 n×mn×m 按照回字蛇形填充至矩阵中。
具体矩阵形式可参考样例。
1.1 输入格式
输入共一行,包含两个整数 nn 和 mm。
1.2输出格式
输出满足要求的矩阵。
矩阵占 nn 行,每行包含 mm 个空格隔开的整数。
1.3数据范围
1≤n,m≤1001≤n,m≤100
1.4输入样例:
3 3
1.5输出样例:
1 2 3
8 9 4
7 6 5
2.解题思路:
这道题的做法有很多种,今天分享一下感觉还不错的方法,采用坐标+1, -1
的形式控制四个方向。看下图即代码讲解
我们首先开一个二维数组,用来存放数字1-n*m
。遍历1-n*m
,然后利用X
和Y
的方向来控制上图中红色箭头 -> 走向,将数字放到数组中正确的位置。
const int N = 110;
int res[N][N];
//x,y坐标
int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0};
// x, y 代表当前坐标
int x = 0, y = 0;
/*
* k从1到n*m, 遍历的目的就是让 k 放到二维数组正确的位置
* d的取值为0-3,为 dx, dy 的下标,用来取dx,dy数组中的值
*/
for(int i = 0, d = 0, k = 1; k <= n*m; k ++)
{
//将k 放入二维数组正确位置
res[x][y] = k;
//用变量 a, b 代表下一个位置,并对下一个位置进行判断是否符合规则
int a = x + dx[d], b = y +dy[d];
//判断 a,b 的值是否符合没有越界,是否已经被访问过
if(a < 0 || a >= n || b < 0 || b >= m || res[a][b])
{
//如果 a,b 值不合法,就进行转弯,利用取模变换坐标
d = (d + 1) % 4;
//更新 a,b 的值
a = x + dx[d], b = y + dy[d];
}
x = a, y = b;
}
3.运行代码
4.完整代码
#include<iostream>
using namespace std;
const int N = 110;
int n,m;
int main()
{
cin >> n >> m;
int res[N][N];
int dx[4] = {0, 1, 0, -1}, dy[4] = {1, 0, -1, 0};
for(int x = 0, y = 0, k = 1, d = 0; k <= n * m; k ++)
{
res[x][y] = k;
int a = x + dx[d], b = y + dy[d];
if(a < 0 || a >= n || b < 0 || b >= m || res[a][b])
{
d = (d + 1) % 4;
a = x + dx[d], b = y + dy[d];
}
x = a, y = b;
}
for(int i = 0; i < n; i ++)
{
for(int j = 0; j < m; j ++)
cout << res[i][j] << ' ';
cout << endl;
}
return 0;
}