국문과 유목민

25. 정수 제곱근 판별 본문

알고리즘_코딩테스트/프로그래머스_Level1

25. 정수 제곱근 판별

논곰 2020. 9. 11. 23:29

0. 문제

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

 

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

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

programmers.co.kr


1. 알고리즘 계획

  1. 제곱근 함수를 이용해서 True일 때는 문제대로 출력, False일 때는 -1출력 (삼항연산자 이용)
  2. 답으로 원하는 값이 정수형이기 때문에 int형변환 시켜준다.


2. 나의 코드

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


3. 다른 사람의 코드

def solution2(n):
    sqrt = pow(n, 0.5) 
    return int(pow(sqrt + 1, 2)) if sqrt == int(sqrt) else -1

- math클래스를 import하지 않고 사용

- 제곱근은 n에 0.5를 제곱한 값임을 이용


4. 정리 및 리뷰

- 배웠던 코드 내에서 생각하여 문제를 풀었다는 점에서 잘 했다고 생각

- 0.5를 제곱하면 제곱근이 된다는 것을 알고 있었는데, 정작 문제를 풀 때는 생각나지 않았었다는 점에 스스로 놀람 


코드 만족도: ★★★


Comments