[프로그래머스] Level1 | 정수 제곱근 판별 - 파이썬(Python) | 연습문제

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

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

프로그래머스에서 레벨1 연습문제를 왕창 풀다가 기억해두면 좋을 것 같은 문제라서 따로 포스팅을 하게 되었다.

정수 제곱급 판별 문제로 math 모듈을 쓰면 엄청 편하다!

---

정답 코드

import math

def solution(n):
    if math.sqrt(n).is_integer() :
        return (math.sqrt(n)+1)**2
    else :
        return -1

 

math.sqrt(x) 함수는 x에 루트를 씌운 값을 반환한다.

반환된 값이 정수인지 아닌지로 판단해서 정수가 아니면 제곱근이 아니기 때문에 -1을 반환고, 정수라면 제곱근이기 때문에 n+1의 제곱을 반환한다.

 

import math를 사용하지 않는 방법도 찾아봤는데 복잡하고 코드도 길어져서 확실히 모듈을 사용하는 것이 좋은 것 같다.

---

해당 문제 풀이 코드

GitHub - YOOHYOJEONG/algorithm_practice: 알고리즘 공부 및 코딩테스트 준비