[Lv.1/파이썬] 프로그래머스 - 정수 제곱근 판별

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

프로그래머스

---

 

내 코드 

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

def solution(n):
    answer = 0
    
    for i in range(1,n+1):
        if i**2 == n:
            answer = i
            break
    
    if answer >= 2:
        return (answer+1)**2
    elif answer == 1:
        return (answer+1)**2
    else:
        return -1

 

다른 사람 코드  

def nextSqure(n):
    from math import sqrt
    return "no" if sqrt(n) % 1 else (sqrt(n)+1)**2

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

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

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

 

 

---

깨달은 점

- math 라이브러리의 sqrt함수(제곱근) 사용하기