[백준][python] 17626 - Four squares

문제

• 백준 17626: Four Squares
• Silver V
• 문제링크

17626번: Four Squares

 

N = int(input())
min_sum = 4 
for a in range(int(N**0.5), int((N//4)**0.5), -1): 
    if a*a == N:
        min_sum = 1 
        break
    else:
        temp = N - a*a
        for b in range(int(temp**0.5), int((temp//3)**0.5), -1): 
            if a*a + b*b == N:
                min_sum = min(min_sum, 2)
                continue
            else:
                temp = N - a*a - b*b
                for c in range(int(temp**0.5), int((temp//2)**0.5), -1):
                    if n == a*a + b*b + c*c:
                        min_sum = min(min_sum, 3)
                
print(min_sum)