[프로그래머스] Lv.0 - 제곱수 판별하기.py

문제

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

제한사항

• 1 ≤ n ≤ 1,000,000

입출력 예시

• 입출력 예 1
  • 144는 12의 제곱이므로 제곱수입니다. 따라서 1을 return합니다.
• 입출력 예 2
  • 976은 제곱수가 아닙니다. 따라서 2를 return합니다.

코드 1

def solution(n):
    answer = 2
    num = 0
    while(num * num < n):
        num += 1
        if num * num == n:
            answer = 1
    return answer

코드 2

def solution(n):
    answer = 2
    if n ** 0.5 == int(n ** 0.5):
        answer = 1
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges