[已解决]重金求方案（

半灵桑 · 发表于 2023-11-20 22:22:34

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

#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)

复制代码

跳转到最佳答案楼层

不二如是 · 发表于 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)

复制代码

歌者文明清理员 · 发表于 2023-11-20 23:04:56

你说的是哪种幻方？

tommyyu · 发表于 2023-11-21 10:08:13

《random》

tommyyu · 发表于 2023-11-21 10:09:45

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

半灵桑 · 发表于 2023-11-21 21:11:21

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

高次幻方

半灵桑 · 发表于 2023-11-21 21:18:42

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

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

tommyyu · 发表于 2023-11-21 21:20:35

本帖最后由 tommyyu 于 2023-11-21 21:24 编辑

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

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

tommyyu · 发表于 2023-11-21 21:25:16

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

半灵桑 · 发表于 2023-11-21 22:18:46

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

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

tommyyu · 发表于 2023-11-22 11:40:39

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

*至少超过一个小时

yinda_peng · 发表于 2023-11-22 15:05:23

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

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

mojo？

tommyyu · 发表于 2023-11-22 21:42:49

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

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

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

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

账号		自动登录	找回密码
密码			立即注册

[已解决]重金求方案（

最佳答案

本帖被以下淘专辑推荐:

浏览过的版块