756蛇形矩阵

题目描述

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

题解思路（平欢）

注意两点：方向的偏移量；撞墙的判断
1、方向偏移 有固定的顺序以及规律 右下左上
2、撞墙判断 a.出界（出数组的边界）; b.遇重 (遇上已经存在数的位置)

C++ 代码

#include <iostream>

using namespace std;

int res[100][100];

int main()
{
    int n, m;
    cin >> n >> m;

    int dx[4] = {0, 1, 0, -1}, dy[4] = {1, 0, -1, 0};
    for (int x = 0, y = 0, d = 0, k = 1; k <= n * m; k ++ ) // x,y 为开始坐标，d 是方向，k 是所赋的值
    {
        res[x][y] = k;

        int a = x + dx[d], b = y + dy[d];  //a,b是当前位置的下一个位置
        if (a < 0 || a >= n || b < 0 || b >= m || res[a][b]) // 撞墙判断： 1、出界；2、遇重
        {
            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;
}