본문 바로가기
알고리즘/문자열, 기본

[프로그래머스 - Level 1] 정수 제곱근 판별(파이썬)

by 호리미 2022. 1. 11.

https://programmers.co.kr/learn/courses/30/lessons/12934

 

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

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

programmers.co.kr

 

<내 코드>

- math.sqrt(x)는 float형으로 결과를 나타냄

import math
def solution(n):
    answer = 0
    x = math.sqrt(n)
    if x - int(x) != 0:
        answer = -1
    else:
        answer = (x+1)**2
    return answer