蛇形矩阵
两个整数n、m,输出一个n行m列的矩阵,将数字 1 到 n*m 按照回字蛇形填充至矩阵中
输入:
3 3
输出:
1 2 3
8 9 4
7 6 5
思路:当前点(x,y),定义方向0、1、2、3代表四个可以走的方向,每次计算下一步会走到的地方,判断是否合法,不合法按顺时针调整方向(用(d+1)%4的结果作为调整后的方向),最终把n*m个数字填入矩阵
#include<iostream>
using namespace std;
const int N = 510;
int f[N][N];
int main()
{
int n, m;
cin >> n >> m;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
int x = 0, y = 0, d = 1; //左上角向右开始,点的当前坐标(x,y)为(0,0),方向d=1
for(int i = 1; i <= n*m; i++)
{
f[x][y] = i; //把当前数字填到当前走到的位置
int a = x + dx[d], b = y + dy[d]; //下一步要走到的坐标
if(a < 0 || a >= n || b < 0 || b >= m || f[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 << f[i][j] << ' ';
cout << endl;
}
return 0;
}