利用python解决魔方阵问题

# 一居上行正中央：数字 1 放在首行最中间的格子中；
# 依次斜填切莫忘：向右上角斜行，依次填入数字；
# 上出框时向下放：如果右上方向出了上边界，就以出框后的虚拟方格位置为基准，将数字竖直降落至底行对应的格子中；
# 右出框时向左放：同上，向右出了边界，就以出框后的虚拟方格位置为基准，将数字平移至最左列对应的格子中；
# 排重便在下格填：如果数字 n右上的格子已被其它数字占领，就将 n+1 填写在 n下面的格子中；
# 右上排重一个样：如果朝右上角出界，和上面重复的情况做同样处理。
def generate_magic_square(n):
    magic_square = [[0] * n for _ in range(n)]
    # 初始位置
    i = 0
    j = n // 2
    num = 1  # 一居上行正中央：数字 1 放在首行最中间的格子中
    while num <= n**2:
        magic_square[i][j] = num
        num += 1
        next_i, next_j = (i - 1) % n, (j + 1) % n  # 依次斜填切莫忘：向右上角斜行，依次填入数字；
        if magic_square[next_i][next_j] != 0:
            i = (i + 1) % n # 排重便在下格填：如果数字 n右上的格子已被其它数字占领，就将 n+1 填写在 n下面的格子中；
        else:
            i, j = next_i, next_j
    return magic_square
def print_magic_square(square):
    for row in square:
        print(" ".join(map(str, row)))
# 生成5行5列的魔方阵并输出
magic_square_5 = generate_magic_square(5)
print_magic_square(magic_square_5)
#该算法优势在于使用了%，可以使得不用考虑出边界的问题