프로그래머스 [코딩테스트 입문 100] 제곱수 판별하기 (Python)

math function을 사용하여 만약 math.sqrt 가 0.5제곱과 같다면 1을 리턴, 아니면 2를 리턴

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