[已解决]重金求方案（

半灵桑

我最近闲着没事，整了个十阶幻方生成器：

#made with python3.10

import random

print('生成中……\n')

while True:
    x = []
    for count in range(10):
        x.append([])

    for count in range(10):
        x[0].append(random.randint(-9999999999, 1000000000))
        x[1].append(random.randint(-9999999999, 1000000000))
        x[2].append(random.randint(-9999999999, 1000000000))
        x[3].append(random.randint(-9999999999, 1000000000))
        x[4].append(random.randint(-9999999999, 1000000000))
        x[5].append(random.randint(-9999999999, 1000000000))
        x[6].append(random.randint(-9999999999, 1000000000))
        x[7].append(random.randint(-9999999999, 1000000000))
        x[8].append(random.randint(-9999999999, 1000000000))
        x[9].append(random.randint(-9999999999, 1000000000))

    a = bool(sum(x[0]) == sum(x[1]) == sum(x[2])== sum(x[3]) == sum(x[4]) \
    == sum(x[5]) == sum(x[6]) == sum(x[7])== sum(x[8]) == sum(x[9]))

    b = bool(x[0][0] + x[0][1] + x[0][2] + x[0][3] + x[0][4] + x[0][5] + x[0][6] + x[0][7] + x[0][8] + x[0][9] == \
             x[1][0] + x[1][1] + x[1][2] + x[1][3] + x[1][4] + x[1][5] + x[1][6] + x[1][7] + x[1][8] + x[1][9] == \
             x[2][0] + x[2][1] + x[2][2] + x[2][3] + x[2][4] + x[2][5] + x[2][6] + x[2][7] + x[2][8] + x[2][9] == \
             x[3][0] + x[3][1] + x[3][2] + x[3][3] + x[3][4] + x[3][5] + x[3][6] + x[3][7] + x[3][8] + x[3][9] == \
             x[4][0] + x[4][1] + x[4][2] + x[4][3] + x[4][4] + x[4][5] + x[4][6] + x[4][7] + x[4][8] + x[4][9] == \
             x[5][0] + x[5][1] + x[5][2] + x[5][3] + x[5][4] + x[5][5] + x[5][6] + x[5][7] + x[5][8] + x[5][9] == \
             x[6][0] + x[6][1] + x[6][2] + x[6][3] + x[6][4] + x[6][5] + x[6][6] + x[6][7] + x[6][8] + x[6][9] == \
             x[7][0] + x[7][1] + x[7][2] + x[7][3] + x[7][4] + x[7][5] + x[7][6] + x[7][7] + x[7][8] + x[7][9] == \
             x[8][0] + x[8][1] + x[8][2] + x[8][3] + x[8][4] + x[8][5] + x[8][6] + x[8][7] + x[8][8] + x[8][9] == \
             x[9][0] + x[9][1] + x[9][2] + x[9][3] + x[9][4] + x[9][5] + x[9][6] + x[9][7] + x[9][8] + x[9][9] )
             

    c = bool(x[0][0] + x[1][1] + x[2][2] + x[3][3] + x[4][4] + x[5][5] + x[6][6] + x[7][7] + x[8][8] + x[9][9] == \
    x[0][9] + x[1][8] + x[2][7] + x[3][6] + x[4][5] + x[5][4] + x[6][3] + x[7][2] + x[8][1] + x[9][0])

    d = bool(x[0][0] != x[0][1] != x[0][2] != x[0][3] != x[0][4] != x[0][5] != x[0][6] != x[0][7] != x[0][8] != \
             x[0][9] != x[1][0] != x[1][1] != x[1][2] != x[1][3] != x[1][4] != x[1][5] != x[1][6] != x[1][7] != \
             x[1][8] != x[1][9] != x[2][0] != x[2][1] != x[2][2] != x[2][3] != x[2][4] != x[2][5] != x[2][6] != \
             x[2][7] != x[2][8] != x[2][9] != x[3][0] != x[3][1] != x[3][2] != x[3][3] != x[3][4] != x[3][5] != \
             x[3][6] != x[3][7] != x[3][8] != x[3][9] != x[4][0] != x[4][1] != x[4][2] != x[4][3] != x[4][4] != \
             x[4][5] != x[4][6] != x[4][7] != x[4][8] != x[4][9] != x[5][0] != x[5][1] != x[5][2] != x[5][3] != \
             x[5][4] != x[5][5] != x[5][6] != x[5][7] != x[5][8] != x[5][9] != x[6][0] != x[6][1] != x[6][2] != \
             x[6][3] != x[6][4] != x[6][5] != x[6][6] != x[6][7] != x[6][8] != x[6][9] != x[7][0] != x[7][1] != \
             x[7][2] != x[7][3] != x[7][4] != x[7][5] != x[7][6] != x[7][7] != x[7][8] != x[7][9] != x[8][0] != \
             x[8][1] != x[8][2] != x[8][3] != x[8][4] != x[8][5] != x[8][6] != x[8][7] != x[8][8] != x[8][9] != \
             x[9][0] != x[9][1] != x[9][2] != x[9][3] != x[9][4] != x[9][5] != x[9][6] != x[9][7] != x[9][8] != x[9][9])

    if a == b == c == d == True :
        break

print('生成完毕！\n')
print(*x[0])
print(*x[1])
print(*x[2])
print(*x[3])
print(*x[4])
print(*x[5])
print(*x[6])
print(*x[7])
print(*x[8])
print(*x[9])

没有BUG，但是效率好低（
完全跑不动啊（
求一个好使的优化方案（

最佳答案

月排行榜 / 总排行榜

不二如是

2023-11-20 22:22:35

利用NumPy的数组操作来计算行和、列和以及对角线和：

import random
import numpy as np

print('生成中……\n')

while True:
    x = np.random.randint(-9999999999, 1000000000, size=(10, 10))

    row_sums = x.sum(axis=1)
    col_sums = x.sum(axis=0)
    diagonal_sum1 = np.trace(x)
    diagonal_sum2 = np.trace(np.fliplr(x))

    if all(row_sums == row_sums[0]) and all(col_sums == col_sums[0]) and diagonal_sum1 == diagonal_sum2:
        break

print('生成完毕！\n')
for row in x:
    print(*row)

跳转到最佳答案楼层

不二如是

利用NumPy的数组操作来计算行和、列和以及对角线和：

import random
import numpy as np

print('生成中……\n')

while True:
    x = np.random.randint(-9999999999, 1000000000, size=(10, 10))

    row_sums = x.sum(axis=1)
    col_sums = x.sum(axis=0)
    diagonal_sum1 = np.trace(x)
    diagonal_sum2 = np.trace(np.fliplr(x))

    if all(row_sums == row_sums[0]) and all(col_sums == col_sums[0]) and diagonal_sum1 == diagonal_sum2:
        break

print('生成完毕！\n')
for row in x:
    print(*row)

歌者文明清理员

你说的是哪种幻方？

tommyyu

《random》

tommyyu

https://www.luogu.com.cn/problem/P2615?contestId=141774

半灵桑

歌者文明清理员 发表于 2023-11-20 23:04
你说的是哪种幻方？

高次幻方

半灵桑

tommyyu 发表于 2023-11-21 10:08
《random》

随机生成100个数字，检验是否符合条件，符合就输出啊？

tommyyu

半灵桑 发表于 2023-11-21 21:18
随机生成100个数字，检验是否符合条件，符合就输出啊？

https://zhuanlan.zhihu.com/p/121838438
你这个程序的时间复杂度 O(m^n) 起步（n -> 幻方大小, m -> 随机范围），不超时才怪

tommyyu

换个方法吧https://blog.csdn.net/tengweitw/article/details/17687377

半灵桑

tommyyu 发表于 2023-11-21 21:20
https://zhuanlan.zhihu.com/p/121838438
你这个程序的时间复杂度 O(m^n) 起步（n -> 幻方大小, m ->  ...

超时？我闲着没事搞的啊（

tommyyu

半灵桑 发表于 2023-11-21 22:18
超时？我闲着没事搞的啊（

*至少超过一个小时

yinda_peng

这种问题的优化……去stack overflow问试试看

歌者文明清理员

mojo？

tommyyu

歌者文明清理员 发表于 2023-11-22 17:41
mojo？

之前有人扒出来，mojo 在官方声明里面写的大意是：

对于所有的 Python3 代码，我们内置了一个 CPython 解释器，并直接用 CPython 运行。

《真·兼容 Python》
详见https://www.bilibili.com/video/B ... id_from=333.999.0.0