[算法基础]蛇形矩阵

蛇形矩阵

#算法笔试面试辅导

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;
}