[파이썬] 프로그래머스 제곱수 판별하기

문제

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

입출력 예

제출한 코드

import math
def solution(n):
    if n % math.sqrt(n) == 0 :
        return 1
    else :
        return 2     

다른사람 풀이

def solution(n):
    if n**(1/2) == int(n**(1/2)) :
        return 1
    else :
        return 2