[프로그래머스] 제곱수 판별하기 - Python

문제 설명

어떤 자연수를 제곱했을 때 나오는 정수를 제곱수라고 합니다. 정수 n이 매개변수로 주어질 때, n이 제곱수라면 1을 아니라면 2를 return하도록 solution 함수를 완성해주세요.

제한사항

1 ≤ n ≤ 1,000,000

입출력 예

---

입출력 예 #1

144는 12의 제곱이므로 제곱수입니다. 따라서 1을 return합니다.

입출력 예 #2

976은 제곱수가 아닙니다. 따라서 2를 return합니다.

---

solution.py

def solution(n):
    answer = 0
    return 1 if (n ** 0.5) == int(n ** 0.5) else 2

def solution(n):
    answer = 0
    return 1 if (n ** 0.5).is_intger() else 2

다른 사람의 풀이

import math

def solution(n):
    return 1 if int(math.sqrt(n)) ** 2 == n else 2