AcWing 756. 蛇形矩阵 ( 螺旋矩阵 )

题目链接：点击进入

题目

思路

模拟螺旋矩阵（蛇形矩阵）

代码

#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<iostream>
#include<cmath>
using namespace std;
const int maxn=1e5+10;
const int dx[4]={0,1,0,-1};
const int dy[4]={1,0,-1,0};
int f[1010][1010];
void init(int n,int m)
{
    int x=1,y=1,d=0;
    for(int i=1;i<=n*m;i++)
    {
        if((x+dx[d]>n||y+dy[d]>m||y+dy[d]<=0)||(f[x+dx[d]][y+dy[d]]))
            d=(d+1)%4;
        f[x][y]=i;
        x+=dx[d];
        y+=dy[d];
    }
}
int main()
{
	ios::sync_with_stdio(false);
    int n,m,r,c;
    cin>>n>>m;
    init(n,m);
    for(int i=1;i<=n;i++)
    {
    	for(int j=1;j<=m;j++)
    	cout<<f[i][j]<<" ";
    	cout<<endl;
	}
	return 0;
}