幻方是一种很神奇的N*N矩阵 问题 A: 神奇的幻方

问题 A: 神奇的幻方
时间限制: 1 Sec 内存限制: 128 MB
提交: 149 解决: 119
[提交][状态][讨论版][命题人:admin]
题目描述
幻方是一种很神奇的N*N矩阵：它由数字1,2,3,……,N*N构成，且每行、每列及两条对角线上的数字之和都相同。
当N为奇数时，我们可以通过以下方法构建一个幻方：
首先将1写在第一行的中间。
之后，按如下方式从小到大依次填写每个数K(K=2,3,…,N*N)：
1.若(K−1)在第一行但不在最后一列，则将K填在最后一行，(K−1)所在列的右一列；
2.若(K−1)在最后一列但不在第一行，则将K填在第一列，(K−1)所在行的上一行；
3.若(K−1)在第一行最后一列，则将K填在(K−1)的正下方；
4.若(K−1)既不在第一行，也不在最后一列，如果(K−1)的右上方还未填数，则将K填在(K−1)的右上方，否则将K填在(K−1)的正下方。
现给定N请按上述方法构造N*N的幻方。
输入
输入只有一行，包含一个整数N即幻方的大小。1≤N≤39且 N 为奇数
输出
输出包含N行，每行N个整数，即按上述方法构造出的N*N的幻方。相邻两个整数之间用单个空格隔开。
样例输入
3
样例输出
8 1 6
3 5 7
4 9 2
提示

纯模拟题。附上代码。

#include <iostream>
#include <cstdio>
#define N 40
using namespace std;
int a[N][N];
int main()  {
    int n,k,k_1i=0,k_1j;
    scanf("%d",&n);
    int dn=n*n;
    k_1j=n/2;
    a[k_1i][k_1j]=1;
    for (k=2;k<=dn;k++) {
        if (k_1i==0&&k_1j!=n-1) {
            a[n-1][k_1j+1]=k;
            k_1i=n-1;
            k_1j=k_1j+1;
        }
        else if (k_1j==n-1&&k_1i!=0)    {
            a[k_1i-1][0]=k;
            k_1i=k_1i-1;
            k_1j=0;
        }
        else if (k_1i==0&&k_1j==n-1)    {
            a[k_1i+1][k_1j]=k;
            k_1i=k_1i+1;
        }
        else if (k_1i!=0&&k_1j!=n-1)    {
            if (a[k_1i-1][k_1j+1]==0)   {
                a[k_1i-1][k_1j+1]=k;
                k_1i=k_1i-1;
                k_1j=k_1j+1;
            }
            else    {
                a[k_1i+1][k_1j]=k;
                k_1i=k_1i+1;
            }
        }
    }
    for (int i=0;i<n;i++)   {
        for (int j=0;j<n;j++)   {
            printf("%d",a[i][j]);
            if (j!=n-1) printf(" ");
            else    printf("\n");
        }
    }
    return 0;
}