Оптимизация кода - Python - Ответ 14970036

def intSqrt(n):
    if n < 2:
        return n
    else:
        sm = intSqrt(n >> 2) << 1
        lg = sm + 1
        if lg * lg > n :
            return sm 
        else:
            return lg 
 
def list_squared(m, n):
    answer=[]
    for i in range(m,n+1):
        k=0
        j = 1
        while j * j < i + 1 :
            if i%(j)==0:
                ij = i // j
                k += (j * j + ij * ij if j - ij else j * j)
            j += 1
        sq = intSqrt(k)
        if sq * sq == k :
            answer.append([i,k])
    return answer
    
print(list_squared(1,250))

from math import sqrt
def list_squared(m, n):
    answer=[]
    for i in range(m,n+1):
        k=0
        j = 1
        while j * j < i + 1 :
            if i%(j)==0:
                ij = i // j
                k += (j * j + ij * ij if j - ij else j * j)
            j += 1
        #sq = intSqrt(k)
        if int(sqrt(k)) * sqrt(k) == k :
            answer.append([i,k])
    return answer
    
print(list_squared(1,25000))