다이어트 [백준 1484]

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

 

import sys

G = int(sys.stdin.readline().strip())

answers = []
low = 1
high = 2 

while True:
    diff = high*high - low*low

    # (diff == G)이면 후보가 됨
    if diff == G:
        answers.append(high)  # 현재 몸무게 = high
    
    if diff >= G:
        low += 1
    else:
        high += 1

    if low >= high:
        break
    if high > 100000:
        break

# 결과 출력
if not answers:
    print(-1)
else:
    # 오름차순 출력
    for ans in sorted(answers):
        print(ans)