제곱수 판별하기
2024. 9. 22. 18:00ㆍ코딩테스트 입문
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
반복문을 돌려서도 문제를 풀 수 있기는 한데, 굳이 그렇게까지 할 필요는 없습니다.