python脚本，对四个五次方进行暴力搜索，四个五次方的总和为五分之一p

Question

在下面的问题中，我试图编写一个python脚本，对四个五次方进行暴力搜索，四个五次方的总和等于五次方。因此，我试图找到数字a1，a2，a3，a4，a5，这样a1^5+a2^5+a3^5+a4^5=a5^5

此搜索的范围是Nmin=20和Nmax=200。另外，Nmin<；=a1<；a2<；a3<；a4<；a5<；=Nmax。所以每个数都必须小于之前的数，并且必须在20到200之间。你知道吗

import numpy as np

b  = np.arange(20,201)
a = b**5

for x in b:
    for y in b:
        for z in b:
            for w in b:

                lambdas=[x,y,z,w]

                if sum(np.array(lambdas)**5) in a:
                    print((x,y,z,w))

其输出如下：

(20, 64, 128, 192)
(20, 64, 192, 128)
(20, 128, 64, 192)
(20, 128, 128, 128)
(20, 128, 192, 64)
(20, 192, 64, 128)
(20, 192, 128, 64)

所以这是错误的输出，我应该得到以下结果：

(27, 84, 110, 133)

似乎没有考虑a1<；a2<；a3<；a4<；a5。我正在寻找一些帮助修复此代码！不是为了提高效率，跑得更快，只是为了得到正确的答案

Answer 1

最重要的是，如果您想拥有这个限制，您需要将它编码到您的循环范围中。只需使用前一个循环的变量作为起始值，就可以得到所需的限制。你知道吗

import numpy as np

lower = 20
upper = 201
b = np.arange(lower, upper)

a = b**5

for x in b:
    for y in np.arange(x, upper):
        for z in np.arange(y, upper):
            for w in np.arange(z, upper):

                lambdas=[x,y,z,w]
                if sum(np.array(lambdas)**5) in a:

                    print((x,y,z,w))

输出：

(27, 84, 110, 133)

我不知道你是怎么运行上一个循环来得到你发布的答案的。你知道吗

你为什么要用numpy作为你的控件？这似乎增加了一层只会减慢处理速度的软件。相反，只需将其编码为“普通”Python数学。另外，使用set检查总和，而不是列表。对集合进行散列也会加快您多次执行的一次检查。你知道吗

lower = 20
upper = 201
b = range(lower, upper)

a = {i**5 for i in b}

for x in b:
    for y in np.arange(x, upper):
        for z in np.arange(y, upper):
            for w in np.arange(z, upper):
                if x**5 + y**5 + z**5 + w**5 in a:
                    print((x,y,z,w))

这得到了正确的答案，并在20秒内完成；您的原始版本（一旦我更正了缩进）花了3分钟。你知道吗