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

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

 

제곱 연산자인 ** 를 사용

 

0.5를 제곱 하면 n 의 제곱근이 float 형태로 반환된다.

n ** 0.5 값을 int로 치환하고 다시 ** 2 (제곱)을 했을 때 주어진 수 n 과 같다면 n은 제곱수 이다.

 

 

파이썬 산술연산자 참조

[develop/python3] - [파이썬/Python] 산술 연산자