AcWing 756. 蛇形矩阵
原题链接
困难
C++ 代码
#include <iostream>
using namespace std;
int main()
{
int n,m,a[110][110],i,j,x=0,y=0,dir=1;//dir表示方向,1右,2下,3左,4上
cin>>n>>m;
for(i=1;i<=n*m;i++) //因为共有n*m个数字要填,所以循环n*m次
{
a[x][y]=i; //x为纵坐标,y为横坐标
switch(dir)
{
case 1:if(y+1<m&&a[x][y+1]==0) y++;else {dir=2;x++;}break;//如果没有越界则按照方向继续,否则转向走
case 2:if(x+1<n&&a[x+1][y]==0) x++;else {dir=3;y--;}break;
case 3:if(y-1>=0&&a[x][y-1]==0) y--;else {dir=4;x--;}break;
case 4:if(x-1>=0&&a[x-1][y]==0) x--;else {dir=1;y++;}break;
}
}
for(i=0;i<n;i++)
{
for(j=0;j<m;j++) cout<<a[i][j]<<' ';
cout<<endl;
}
}