[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())

data = []
for i in range(1, n + 1):
    if m <= i ** 2 <= n:
        data.append(i ** 2)

if len(data) == 0:
    print(-1)
else:
    print(sum(data))
    print(data[0])

우선, 완전제곱수란 자연수를 제곱하여 만들어지는 수를 의미합니다.

math를 사용해서 풀어보려다가 직관적으로 풀었습니다. m과 n 사이의 완전제곱수를 찾기 위해 대소 비교를 해주어 리스트에 추가했습니다. 리스트가 비어있는 경우는 m과 n 사이에 완전제곱수가 존재하지 않는다는 의미이기 때문에 문제에 제시된대로 -1을 출력합니다.

 

피드백은 언제나 환영합니다.