[NOIP2015 提高组] 神奇的幻方

题目描述

幻方是一种很神奇的 N*NN∗N 矩阵：它由数字 1,2,3,\cdots \cdots ,N \times N1,2,3,⋯⋯,N×N 构成，且每行、每列及两条对角线上的数字之和都相同。

当 NN 为奇数时，我们可以通过下方法构建一个幻方：

首先将 11 写在第一行的中间。

之后，按如下方式从小到大依次填写每个数 K (K=2,3,\cdots,N \times N)K(K=2,3,⋯,N×N) ：

1. 若 (K-1)(K−1) 在第一行但不在最后一列，则将 KK 填在最后一行， (K-1)(K−1) 所在列的右一列；
2. 若 (K-1)(K−1) 在最后一列但不在第一行，则将 KK 填在第一列， (K-1)(K−1) 所在行的上一行；
3. 若 (K-1)(K−1) 在第一行最后一列，则将 KK 填在 (K-1)(K−1) 的正下方；
4. 若 (K-1)(K−1) 既不在第一行，也不在最后一列，如果 (K-1)(K−1) 的右上方还未填数，则将 KK 填在 (K-1)(K−1) 的右上方，否则将 KK 填在 (K-1)(K−1) 的正下方。

现给定 NN ，请按上述方法构造 N \times NN×N 的幻方。

输入格式

一个正整数 NN ，即幻方的大小。

输出格式

共 NN 行 ，每行 NN 个整数，即按上述方法构造出的 N \times NN×N 的幻方，相邻两个整数之间用单空格隔开。

对于100\%100%的数据，对于全部数据， 1 \leq N \leq 391≤N≤39 且 NN 为奇数。

 输入 #1                                                         

3                                    
输出 #1
8 1 6
3 5 7
4 9 2

 输入 #2

25

 

water 模拟 ~

提高组混了个普及题（doge）给我逮到了吧~

#include<bits/stdc++.h>
#define pre(i,a,b) for(int i=a;i<=b;i++)
#define rep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
using namespace std;
const int maxn=1e6+5;
int vis[40][40];
int main()
{
	int n;
	cin>>n;
	vis[0][n/2]=1;
	int h=0,l=n/2;
	pre(i,2,n*n){
		if(h==0&&l!=n-1) vis[n-1][l+1]=i,h=n-1,l++;
		else if(l==n-1&&h!=0) vis[h-1][0]=i,h--,l=0;
		else if(h==0&&l==n-1) vis[h+1][l]=i,h++;
		else if(h!=0&&l!=n-1){
			if(vis[h-1][l+1]==0&&h-1>=0&&l+1<=n) vis[h-1][l+1]=i,h--,l++;
			else vis[h+1][l]=i,h++;
		}
	}
	pre(i,0,n-1){
		pre(j,0,n-1)cout<<vis[i][j]<<' ';
		cout<<endl;
	}
}