[프로그래머스] Level 1_ 정수 제곱근 판별

유형

연습문제

 

문제

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

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

 

나의 문제풀이1 (실패)

def solution(n):
    for x in range(1,n//2+1):
        if x**2 == n:
            return (x+1)**2
        else:
            return -1

 

나의 문제풀이2 (실패)

 - try,except, else 문 https://blockdmask.tistory.com/537

[python] 파이썬 예외처리 try, except, else, finally 사용 방법

import math as m

def solution(n):
    try:
        int(m.pow(n)) == False
    except:
        return -1
    else:
        return (n+1)**2

 

나의 문제풀이3 (실패 - 정확성 94.4)

def solution(n):
    for x in range(1,n//2+1):
        if x ** 2 == n:
            return (x+1)**2
    return -1

 

나의 문제풀이4 (성공)

- 실패 원인 : for문 돌아가는 횟수때문에 속도가 느렸음

 - 해결방법 : 제곱근 횟수만큼 for문 사용 후 해결

import math as m

def solution(n):
    for x in range(1,int(m.sqrt(n))+1):
        if x ** 2 == n:
            return (x+1)**2
    return -1

 

다른 사람들의 문제풀이

 1) **(1/2) 사용

def nextSqure(n):
    sqrt = n ** (1/2)

    if sqrt % 1 == 0:
        return (sqrt + 1) ** 2
    return -1

 

 2) 1번 한 줄 코드

def nextSqure(n):
    return n == int(n**.5)**2 and int(n**.5+1)**2 or -1

 

 3) sqrt와 is_integer() 사용

• is_integer() : float 객체에서 정수인지 핀별
  https://www.delftstack.com/ko/howto/python/is-int-or-not-python/#float.is_integer%25EB%25A9%2594%25EC%2586%258C%25EB%2593%259C%25EB%25A5%25BC-%25EC%2582%25AC%25EC%259A%25A9%25ED%2595%2598%25EC%2597%25AC-python%25EC%2597%2590%25EC%2584%259C-%25EA%25B0%259D%25EC%25B2%25B4%25EA%25B0%2580int%25EC%259C%25A0%25ED%2598%2595%25EC%259D%25B8%25EC%25A7%2580-%25ED%2599%2595%25EC%259D%25B8

변수가 정수 Python인지 확인

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

 

 4) pow 이용

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