Bạn có thể đơn giản hóa số học bằng cách sử dụng
[n + 1]**2 == n**2 + [2*n + 1]
Đây là cách thực hiện điều đó bằng cách sử dụng hàm máy phát:
import math
def squares[lo, hi]:
root = int[math.ceil[lo ** 0.5]]
num = root ** 2
delta = 2 * root + 1
while num