[브론즈1]백준 1977번 완전제곱수 - Python

1977번 완전제곱수

사이트 바로가기 : 1977번 완전제곱수

문제

M과 N이 주어질 때 M이상 N이하의 자연수 중 완전제곱수인 것을 모두 골라 그 합을 구하고 그 중 최솟값을 찾는 프로그램을 작성하시오. 예를 들어 M=60, N=100인 경우 60이상 100이하의 자연수 중 완전제곱수는 64, 81, 100 이렇게 총 3개가 있으므로 그 합은 245가 되고 이 중 최솟값은 64가 된다.

첫째 줄에 M이, 둘째 줄에 N이 주어진다. M과 N은 10000이하의 자연수이며 M은 N보다 같거나 작다.

M이상 N이하의 자연수 중 완전제곱수인 것을 모두 찾아 첫째 줄에 그 합을, 둘째 줄에 그 중 최솟값을 출력한다. 단, M이상 N이하의 자연수 중 완전제곱수가 없을 경우는 첫째 줄에 -1을 출력한다.

풀이

M = int(input())
N = int(input())

import math
min_num = math.ceil(M ** (1 / 2))
max_num = math.trunc(N ** (1 / 2))

answer = []
while True:
    if min_num > max_num:
        print(-1)
        break
        
    else:
        answer.append(min_num ** 2)
        min_num += 1
        if min_num > max_num:
            print(sum(answer))
            print(answer[0])
            break