백준 알고리즘 15667번 2018 연세대학교 프로그래밍 경진대회(python)

단순 사칙연산 문제이다.

K = int(input())
for i in range(1, K+1):
    if 1 + i + i**2 == K:
        print(i)
        break