Lv.0 제곱수 판별하기

문제 설명

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

입출력

코드 풀이

import math

def solution(n):
    sqrt_n = math.sqrt(n)  # n의 제곱근을 계산
    if int(sqrt_n) == sqrt_n:
        return 1  # 제곱근이 정수인 경우, n은 제곱수
    else:
        return 2  # 제곱근이 정수가 아닌 경우, n은 제곱수가 아님

출처: 프로그래머스 코딩테스트 연습