牛客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(" ")

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页