蛇形矩阵

蛇形矩阵

题目描述

输入两个整数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

思路

本质上来说是要输出n * m个数，但是需要转变方向，假如一个点的坐标为（0，0），那么它的上右下左坐标的偏移量为（0，1），（1，0），（0，-1），（-1，0）。假设当前坐标点为（x，y），那么它下一个点的坐标是什么呢？我们首先需要判断模式，什么模式呢？就是它下一个点是往哪个方向走，判断完方向后，下一个点的坐标即为（x，y）+ （该模式下的偏移量，前文已经定义）。初始状态为朝右走，什么时候需要向下走呢？这时候需要考虑边界问题，就是索引不能小于0，不能大于n/m，且该索引对应的值未被填充过。即当

a < 0 || a >= n || b < 0 || b >= m || N[a][b]

条件发生时，转变方向，右→下，下→左，左→上，上→右

C++代码

#include <iostream>

using namespace std;

int N[100][100];

int main()
{
    int n, m;
    cin >> n >> m;
    int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0};
    for (int x = 0, y = 0, d = 0, i = 1; i <= n * m; i ++)
    {
        N[x][y] = i;
        int a = x + dx[d], b = y + dy[d];
        if (a < 0 || a >= n || b < 0 || b >= m || N[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 << N[i][j] << ' ';
        cout << endl;
    }

    return 0;
}

题目来源：756. 蛇形矩阵

真心推荐AcWing网站，无论是课程还是学习分为对我这种小白都很友好。