AcWing756.蛇形矩阵

题目

分析

对于这类问题，我们采用dx, dy 方向数组来存储，并且将顺时针方向定义为1，2，3，0

撞墙的情况有两种，一个是出去，一个是走重复的格子

 

 

代码

#include<iostream>
using namespace std;

const int N = 110;
int n,m;
int q[N][N];

int main(){
    cin>> n >> m;
    
    //定义开始状态，转移增量，方向
    int x = 0,y = 0;
    int dx[4] = {-1,0,1,0},dy[4] = {0,1,0,-1};
    int d = 1;//表示顺时针的四个方向，开始为1横着走
    
    
    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;
    }
}