제곱수 판별하기

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

프로그래머스

---

문제 설명
어떤 자연수를 제곱했을 때 나오는 정수를 제곱수라고 합니다. 정수 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합니다.

---

정수 n이 주어질 때, n이 제곱수인지 판별하는 문제입니다.

 

파이썬에서는 제곱을 **으로 표현하는데요, 여기서 **(1/2)를 한다면 제곱수를 구할 수 있습니다.

따라서 제곱수인지 판별하려면 **(1/2)한 결과가 정수인지만 확인하면 됩니다.

파이썬으로 문제를 풀어보겠습니다.

def solution(n):
    # n의 제곱근을 구합니다.
    num = n ** (1/2)
    # 정수로 변환했을 때 값이 같다면
    if num == int(num):
        # 제곱수이므로 1을 반환합니다.
        return 1
    # 아니라면 2를 반환합니다.
    return 2

 

(1/2) 제곱을 사용해서 제곱근을 구한 다음, 해당 제곱근을 정수형으로 변환했을 때 동일한 값을 갖는지 확인하면 됩니다.

동일한 값이면 1을, 아니라면 2를 반환합니다.

int 말고 is_integer()을 사용해서 정수형으로 변환하셔도 동일한 결과를 출력합니다.

 

한 줄로 표현하면 다음과 같습니다.

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

 

반복문을 돌려서도 문제를 풀 수 있기는 한데, 굳이 그렇게까지 할 필요는 없습니다.