每日一题——蛇形矩阵

问题描述

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

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

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

输出格式
输出满足要求的矩阵。

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

数据范围
1≤n,m≤100
输入样例：

3 3

输出样例：

1 2 3
8 9 4
7 6 5

分析

代码

#include<iostream>
using namespace std;

const int N = 110;

int n,m;
int q[N][N];

int main()
{
    cin>>n>>m;
    int dx[] = {-1,0,1,0},dy[] = {0,1,0,-1};
    int x = 0,y = 0,d = 1;//d代表方向
    
    for(int i = 1;i<=n*m;i++){
        q[x][y] = i;
        int a = x + dx[d],b = y+dy[d];
        // 判断是否出界、或者已经计算过
        if(a<0 || a>=n || b<0 || b>=m || q[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<<q[i][j]<<' ';
        }
        cout<<endl;
    }
    
    return 0;
}