프로그래머스(Python) Level.1 정수 제곱근 판별 풀이

Python으로 구현한 12934번 정수 제곱근 판별 문제 풀이입니다.

 

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

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

---

import math

def solution(n):
    data = math.sqrt(n)
    if data == int(data) :
        return (data + 1) ** 2
    
    return -1

 

1. math.sqrt()를 활용하여 n의 제곱근을 구해 data에 할당한다.

 

2. 만약 data의 값이 정수형으로 변환한 값과 같다면 data + 1의 제곱을 리턴하고, 그렇지 않다면 -1을 리턴한다.