每日一题，幻方模拟

题目描述

幻方是一种很神奇的 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 ，即幻方的大小。

未优化code

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int main(){
    int n;
    scanf("%d",&n);
    int a1[n+5][n+5]={0};
    a1[1][(n+1)/2]=1;
    int a=1,b=(n+1)/2;
    
    for(int i=2;i<=n*n;i++)
    {
    
        if(a==1&&b!=n)
        {
            a1[n][b+1]=i;
            a=n;
            b+=1;
        }
        else if(a!=1&&b==n)
        {
            a1[a-1][1]=i;
            a-=1;
            b=1;
        }
         else if(a==1&&b==n)
        {
            a1[a+1][b]=i;
            a+=1;
            b=b;
        }
    else    if(a!=1&&b!=n)
        {
            if(a1[a-1][b+1]==0)
            {
                a1[a-1][b+1]=i;
                a-=1;
                b=b+1;
            }
            else 
            {
                a1[a+1][b]=i;
                a=a+1;
                b=b;
            }
        }
    }    
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            printf("%d ",a1[i][j]);
        }
        if(i<n)
        {
            printf("\n");
        }
    }
    return 0;
}