[Python / Level1] 정수 제곱근 판별

Question

https://school.programmers.co.kr/learn/courses/30/lessons/12934

프로그래머스

---

Code

def solution(n):
    for i in range(1, n + 1):
        if i ** 2 == n:
            return (i + 1) ** 2
    return -1

---

Idea

어느 수까지 반복하라는 명확한 기준이 없지만 적어도 1~n까지의 특정 수를 제곱하면 n이 나온다는 것을 알 수 있다.

이에 따라 for loop를 실행하며 특정 수를 찾고, +1을 한 수의 제곱을 return하면 되는 문제였다.