프로그래머스 제곱수 판별하기

💡문제 설명

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

---

🚫제한사항

• 1 ≤ n
  ≤ 1,000,000

---

🔢입출력 예

n	result
144	1
976	2

---

🔍입출력 예 설명

입출력 예 #1

• 144는 12의 제곱이므로 제곱수입니다. 따라서 1을 return합니다.

입출력 예 #2

• 976은 제곱수가 아닙니다. 따라서 2를 return합니다.

---

💻코드

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

https://school.programmers.co.kr/learn/courses/30/lessons/120909?language=python3