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

1.문제

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

프로그래머스

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

 

2. 제한사항

 

3. Python code 

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

주어진 숫자(n)을 루트n해서 정수면 n을 제곱수라고 판별해 return1한다.

 

    if (n**(0.5)).is_integer():
        return 1
    else:
        return 2

이 코드는 is_interger()를 이용해 바로 n이 제곱수인지 바로 판별한다.