정수 제곱근 판별

2024. 3. 30. 13:02프로그래머스-코딩테스트/Lv.1

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

문제


문제 설명

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

 

 

 

 

코드


import math as m

def solution(n):
    x = m.sqrt(n)
    if x % int(x) == 0:
        return (x+1) ** 2
    else:
        return -1

 

 

 

 

노트


def solution(n):
    x = n ** .5
    return -1 if x%1 else (x+1)**2
  • .5 를 0.5로 알아듣는 똑똑한 파이썬
  • 1로 나누어 나머지가 없다면, 정수형태인것

 

 

'프로그래머스-코딩테스트 > Lv.1' 카테고리의 다른 글

하샤드 수  (0) 2024.04.07
정수 내림차순으로 배치하기  (0) 2024.04.02
문자열 내 p와 y의 개수  (0) 2024.03.30
약수의 합  (1) 2024.03.26
문자열을 정수로 바꾸기  (0) 2024.03.26