본문 바로가기
알고리즘/프로그래머스 PS

[프로그래머스][Level 1][Python] 정수 제곱근 판별

by 빛밤하늘 2021. 7. 11.
반응형

밑의 링크는 프로그래머스에서의 문제 링크입니다.

 

코딩테스트 연습 - 정수 제곱근 판별

임의의 양의 정수 n에 대해, n이 어떤 양의 정수 x의 제곱인지 아닌지 판단하려 합니다. n이 양의 정수 x의 제곱이라면 x+1의 제곱을 리턴하고, n이 양의 정수 x의 제곱이 아니라면 -1을 리턴하는 함

programmers.co.kr

 

 

 

 

 

 

 

 

파이썬 코드와 결과입니다.

# 양의 정수 n이 매개변수로 주어집니다.
# n은 1 이상, 50,000,000,000,000 이하인 양의 정수입니다.
def solution(n):
    # x+1의 제곱 혹은 -1을 저장할 변수를 선언합니다.
    answer = 0
    # 양의 정수 n의 제곱근을 저장하는 변수를 선언합니다.
    x = n ** 0.5

    # n의 제곱근 x가 .0으로 끝난다면
    if str(x)[-2:] == '.0':
        # x는 양의 정수이므로 x+1의 제곱을 answer에 저장합니다.
        answer = (x + 1) ** 2
    # n의 제곱근 x가 .0으로 끝나지 않는다면
    else:
        # x는 양의 정수가 아니므로 answer에 -1을 저장합니다.
        answer = -1

    # answer의 값을 반환합니다.
    return answer

결과

 

 

 

※ 궁금한 부분, 이상한 점 및 오타는 댓글에 부탁드립니다.

※ 더 효율적이고 빠른 정답을 환영합니다.

반응형

댓글