[Python/BOJ] 1484. 다이어트

https://www.acmicpc.net/problem/1484

1484번: 다이어트

문제

성원이는 다이어트를 시도중이다. 성원이는 정말 정말 무겁기 때문에, 저울이 부셔졌다. 성원이의 힘겨운 다이어트 시도를 보고만 있던 엔토피아는 성원이에게 새로운 저울을 선물해 주었다. 성원이는 엔토피아가 선물해준 저울 위에 올라갔다. “안돼!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! G 킬로그램이나 더 쪘어ㅜㅠ”라고 성원이가 말했다. 여기서 말하는 G킬로그램은 성원이의 현재 몸무게의 제곱에서 성원이가 기억하고 있던 몸무게의 제곱을 뺀 것이다.

성원이의 현재 몸무게로 가능한 것을 모두 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 G가 주어진다. G는 100,000보다 작거나 같은 자연수이다.

출력

첫째 줄부터 한 줄에 하나씩 가능한 성원이의 현재 몸무게를 오름차순으로 출력한다. 가능한 몸무게가 없을 때는 -1을 출력한다. 현재 몸무게는 자연수로 떨어지지 않을 수도 있는데, 이런 경우는 제외해야 한다.

풀이

Code

import sys, math
input = sys.stdin.readline

g = int(input())
ans = []
# 1. 기억하고 있는 몸무게 설정
remember = 1
# 2.
while True :
    # 2-1. 식이 맞아 떨어지는 경우 출력 리스트에 값 삽입
    calculation = math.sqrt(g + remember**2)
    if int(calculation) == calculation : ans.append(int(calculation))
    # 2-2. 정수화시킨 계산값과 기억하는 몸무게가 같을 경우 탈출
    if int(calculation) == remember : break
    # 2-3. 기억하는 몸무게 값 추가
    remember += 1
# 3. 결과 출력
if not ans : print(-1)
else :
    for num in ans : print(num)