蛇形填数是一种二维数组模拟填充,由于填充的起始位置的不同,可以有多种蛇形填数的变种,但是有一种通用方法来解决这些变种。
首先创建一个元素均初始化为0的二维数组,然后我们定义一个起始坐标x和y并对其赋值,接着依次向下向左向上向右(根据题目条件)进行填充,在填充的时候,需要先通过是否为0来进行预判再对其赋值,同时注意填充时的边界限制。
#include<iostream>
#include<vector>
using namespace std;
int main()
{
int n;
cin>>n;
vector<vector<int>> a(n,vector<int>(n,0));
int x=0,y=n-1; //起始位置
int count=1; //用于计数
a[x][y]=count;
while(count<n*n)
{
//通过预判减少回溯
while(x+1<n&&a[x+1][y]==0) //向下
a[++x][y]=++count;
while(y-1>=0&&a[x][y-1]==0) //向左
a[x][--y]=++count;
while(x-1>=0&&a[x-1][y]==0) //向上
a[--x][y]=++count;
while(y+1<n&&a[x][y+1]==0) //向右
a[x][++y]=++count;
}
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
cout<<a[i][j]<<" ";
}
cout<<endl;
}
return 0;
}