[programmers] 프로그래머스 정수 제곱근 판별(Python)

(주)그렙

[programmers] 프로그래머스 정수 제곱근 판별

(Python)

(글쓴날 : 2020.06.13)

 

---

* programmers, 프로그래머스 문제 Python 언어 풀이입니다.

* 소스 코드의 저작권은 글쓴이에게 있습니다.

---

 

 

프로그래머스 정수 제곱근 판별

---

1) 문제

문제 링크 : https://programmers.co.kr/learn/courses/30/lessons/12934

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

---

2) 풀이 과정

* 시간 복잡도 : O(1)

 

양의 정수 n이 주어질 때, 만약 n의 어떤 양의 정수 x의 제곱이라면 x+1의 제곱을 반환하고 그렇지 않다면 -1을 반환하는 문제입니다.

 

저의 경우, Python을 사용했습니다.

먼저 math 라이브러리를 import 한 뒤, sqrt() 함수를 사용해 루트화 한 n이 정수일 경우 해당 값+1을 반환하고, 소수일 경우 -1을 반환하여 문제를 해결했습니다.

---

3) 코드

 

* Python 코드

import math
def solution(n):
    if math.sqrt(n) % 1 == 0:
        return (math.sqrt(n)+1)**2
    
    return -1

---