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

//P5731 【深基5.习6】蛇形方阵
//#define LOCAL
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#define inf 0x3f3f3f3f
#define eps 1e-6
using namespace std;
#define clr(x) memset(x,0,sizeof((x)))
const int maxn = 2e5+1;//2e6+1
#define MAX(a,b,c) ((a)>(b)?((a)>(c)?(a):(c)):((b)>(c)?(b):(c)))
#define _max(a,b) ((a) > (b) ? (a) : (b))
#define _min(a,b) ((a) < (b) ? (a) : (b))
#define _for(a,b,c) for(int a = b;a<c;a++)
int main() {
#ifdef LOCAL 
	freopen("data.in","r",stdin);
	freopen("data.out","w",stdout);
#endif
	int n,x,y,tot = 1;
	cin>>n;
	int buf[10][10];
	clr(buf);
	buf[x = 0][y = 0] = 1;//初始化
	while(tot<n*n) {
		while(y+1<n&&!buf[x][y+1]) buf[x][++y] = ++tot;//右
		while(x+1<n&&!buf[x+1][y]) buf[++x][y] = ++tot;//下
		while(y-1>=0&&!buf[x][y-1]) buf[x][--y] = ++tot;//左
		while(x-1>=0&&!buf[x-1][y]) buf[--x][y] = ++tot;//上
	}
	for(x = 0;x<n;x++) {
		for(y = 0;y<n;y++) {
			printf("%3d",buf[x][y]);
		}
		printf("\n");
	}
	return 0;
}