gpt4 book ai didi

python - 优化 julia one-liner 使其与 python 一样快

转载 作者:太空狗 更新时间:2023-10-30 01:00:14 25 4
gpt4 key购买 nike

我在 julia 中写了一个简单的单行代码来解决一个小数学问题:找到一个两位数 A 和一个三位数 B,使得它们的乘积 A x B 是一个五位数,并且每个数字来自0 到 9 在数字 A、B 和 A x B 中只出现一次。例如,

54 x 297 = 16,038

这是我的 julia 代码,它找到了所有可能的解决方案:

println(filter(l -> length(unique(reduce(vcat, (map(digits, l))))) == 10, [[x, y, x*y] for x in Range(10:99), y in Range(100:999)]))

它解决了问题,但后来我在 python 中尝试并想出了这个:

print filter(lambda y: len(set(''.join([str(x) for x in y])))==10, [[x, y, x*y] for x in range(10, 99) for y in range(100, 999)])

对它们进行计时,我惊讶地发现 python 代码的运行速度是 julia 代码的两倍多。对于 julia 代码的更快方法有什么建议(最好将其保持在一行中)?

另外:我知道我可以通过将范围快速调整到 range(12, 98) 范围(102、987)

更新

超越单行代码,我采纳了循环比列表更快的建议,因此我比较了以下替代方案:

Julia

ans = Array{Tuple{Int32, Int32, Int32}}(0)
for x in 12:98
for y in 102:987
if length(unique(digits(x+y*100+x*y*100_000)))==10 push!(ans, (x, y, x*y) end
end
end
println(ans)

python

ans = []
for x in range(12,98):
for y in range(102,987):
if len(set(str(x+y*100+x*y*100000)))==10:
ans.append((x, y, x*y))
print ans

python 代码运行得更快(即使我更改了两者的代码以简单地在循环中打印出结果而不是将它们收集在列表中)。我期待 julia 有更好的表现。

此外,如果您有兴趣,完整的解决方案列表是

39 x 402 = 15,678
27 x 594 = 16,038
54 x 297 = 16,038
36 x 495 = 17,820
45 x 396 = 17,820
52 x 367 = 19,084
78 x 345 = 26,910
46 x 715 = 32,890
63 x 927 = 58,401

最佳答案

@simd for x in 10:99 for y in 100:999 length(unique(digits(x+y*100+x*y*100_000)))==10 && println(x,'* ',y,'=',x*y) 结束结束

在我的电脑中,这段代码的速度大约是原始代码的 3 倍。 (0.223902 秒对 0.680781 秒)

关键是"avoid unnecessary arrays" .尽可能使用 for 循环或元组

关于python - 优化 julia one-liner 使其与 python 一样快,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36694459/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com