穷举+hash解一个数论问题

$$a^4+b^4+c^4+d^4 = e^4 ...... 1=<a,b,c,d,e<=1000,求所有满足方程的解。$$

这个问题如果直接来一个四重循环穷举是不行的….$1000^4 = 10000亿，以python的速度得跑上不知道多久..$，所以我采取的办法是先预处理1-1000的四次方，然后通过一个二重循环计算出所有$i^4+j^4=>(i,j)且存储起来$,然后再使用一个三重循环来判定是否$e^4-(c^4+d^4)$是否在记录里面,如果在就可以输出一组解,因为$(c^4+d^4)$也是记录里面的$(c,d)$,所以可以用一个二重循环代替三重循环(第二重直接枚举记录就好),当然复杂度不变,只是效率会稍微高一点。

max=1000
x = [0]*(max+5)
def com(n):
    for i in range(1,n+1):
        x[i] = i**4
com(max)
def solve():
    n = int((max**4-3)**(1/4))
    s = {}
    for i in range(1,n+1):
        for j in range(i,n+1):
            k = x[i]+x[j]
            if k in s:s[k]+=(i,j)
            else:s[k]=(i,j)
    u = list(s.keys())
    u.sort()
    for N in range(3,max):
        N = N**4
        for k in u:
            if (N-k) in s:print(s[k],s[N-k],int(N**(0.25)));#s.pop(k,None)
            if k>N-k:break
        #print(N**(1/4))
solve()