洛谷P5731 【深基5.习6】蛇形方阵

#include <bits/stdc++.h>
using namespace std;
int num[100][100];
int main()
{
	int n;	cin>>n;
	num[1][1]=1;//初始化
	for(int i=1,j=1,ans=1;ans<n*n;)
	{
		//向右 
		while(++j<=n&&!num[i][j]){
			num[i][j]=++ans;
		}
		j--;
		//向下
		while(++i<=n&&!num[i][j]){
			num[i][j]=++ans;
		}
		i--;
		//向左
		while(--j>0&&!num[i][j]){
			num[i][j]=++ans;
		}
		j++; 
		//向上
		while(--i>0&&!num[i][j]){
			num[i][j]=++ans;
		} 
		i++;
	}
	for(int i=1;i<=n;i++,cout<<endl)
	{
		for(int j=1;j<=n;j++)
		{
			cout<<setw(3)<<num[i][j];
		}
	}

	return 0;
}