Check square number

해당 문제는 여기에서 확인하실 수 있습니다.

import math

def solution(n):
  # n의 제곱근을 계산합니다.
  sqrt_n = math.sqrt(n)
  
  # 제곱근이 정수라면, n은 제곱수입니다.
  if sqrt_n == int(sqrt_n):
    # n이 제곱수일 경우 1을 반환합니다.
    return 1
  else:
    # n이 제곱수가 아닐 경우 2를 반환합니다.
    return 2