面试题-蛇形矩阵详解

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

//      y  y  y     右(0,1) x不变，y加1
//   x				下(1,0)  x加1，y不变
//	 x	            左(0,-1)  x不变，y减一
//   x              上(-1,0)   x减1，y不变
//1.方向偏移  有固定的顺序以及规律 右下左上
//2.走到头 a.出界（出数组的边界）; 
//         b.走到重复格子 (遇上已经存在数的位置)
#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}; 
    //4可以省略
    for(int x=0,y=0,d=0,k=1;k<=n*m;k++)
  //从左上角开始（0,0） 填的数字是K，k从1开始，一共填n*m个数
    {
        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]) //res[a][b]走到头
        {
            d=(d+1)%4;//循环0-1-2-3-0
            //0变成1   1变成2  2变成3  3变成0 循环
            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;
}
//每个格子走一遍所以时间复杂度 O(n*m)  线性的