level0_제곱수 판별하기

2024. 10. 22. 21:11파이썬_알고리즘(코딩테스트)/level0

 

내가 제출한 코드

def solution(n):
    if (n ** 0.5) % 1 == 0:
        return 1
    else:
        return 2
  • (n  ** 0.5) % 1 == 0 
    • n ** 0.5는 n의 제곱근을 계산하는 표현식이다.
    • 제곱근에 1로 나누었을때 나머지가 0이라면 (n의 제곱근이 정수라면)
    • 1을 반환한다.
  • n의 제곱근이 정수가 아니라면 2를 반환한다.

 

 

다른 사람이 제출한 코드 

def solution(n):
    return 1 if (n ** 0.5).is_integer() else 2
  • is_integer() 메서드를 사용해서 n의 제곱근이 정수이면 1을 반환하고, 아니라면 2를 반환한다.

 

is_integer() 오... 메서드 기억해둬야겠다.