Представление в виде суммы

n = int(input())
lim = int(n**0.5) + 1
flt = []
for x in range(1, lim):
    for y in range(1, lim):
        for z in range(1, lim):
            if n == x**2 + y**2 + z**2 and x not in flt and y not in flt and z not in flt:
                print(x, y, z)
                flt.append(x)
                flt.append(y)
                flt.append(z)

n = 54
lim = int(n**0.5) + 1
flt = []
for x in range(1, lim):
    for y in range(x, lim):
        for z in range(y, lim):
            if n == x**2 + y**2 + z**2 :
                print(x, y, z)

n = int(input())
lim = int(n**0.5) + 1
squares=[(i*i,i) for i in range(1,lim+1)]
sz=len(squares)
res=[]
for i in range(sz):
    for j in range(i,sz):
        for k in range(j,sz):
            (a,ai)=squares[i]
            (b,bj)=squares[j]
            (c,ck)=squares[k]
            if a+b+c==n:
                res.append((ai,bj,ck))
print(res)

n = 10**6
square_roots = {}
i=1
while i*i <= n:
    square_roots[i*i] = i
    i += 1
x = 1
while x * x <= n//3:
    y = x
    while y * y <= (n - x*x)//2:
        if (z:=square_roots.get(n-x*x-y*y)):
                print(x, y, z)
        y += 1
    x += 1