牛客16464 神奇的幻方

链接：https://ac.nowcoder.com/acm/problem/16464
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 131072K，其他语言262144K
64bit IO Format: %lld

题目描述

幻方是一种很神奇的 N*N 矩阵：它由数字 1,2,3,…N x N 构成，且每行、每列及两条对角线上的数字之和都相同。
当 N 为奇数时，我们可以通过下方法构建一个幻方：
首先将 1 写在第一行的中间。
之后，按如下方式从小到大依次填写每个数 K (K=2,3,…,N x 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) 的右上方，否则将 L 填在 (K-1) 的正下方。

输入描述:

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

输出描述:

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

按照题目描述直接分情况讨论就可以

N = int(input())
H = []
for i in range(N):
    H.append([])
    for j in range(N):
        H[i].append(0)

H[0][int((N-1)/2)] = 1
for k in range(2, N**2+1):
    for i in range(N):
        for j in range(N):
            if k-1 == H[i][j]:
                if i==0 and j!=N-1:
                    H[N-1][j+1]=k
                elif i!=0 and j==N-1:
                    H[i-1][0]=k
                elif i==0 and j==N-1:
                    H[i+1][j]=k
                elif i!=0 and j!=N-1:
                    if H[i-1][j+1]==0:
                        H[i-1][j+1]=k
                    else:
                        H[i+1][j]=k
#print(H)

for i in range(N):
    for j in range(N):
        print(H[i][j], end =" ")
    print(" ")