초보몽키의 개발공부로그 level 1. 정수제곱근판별하기
06 Mar 2017 | 알고리즘 프로그래밍level 1. 정수제곱근판별하기
문제
nextSqaure함수는 정수 n을 매개변수로 입력받습니다. n이 임의의 정수 x의 제곱이라면 x+1의 제곱을 리턴하고, n이 임의의 정수 x의 제곱이 아니라면 ‘no’을 리턴하는 함수를 완성하세요. 예를들어 n이 121이라면 이는 정수 11의 제곱이므로 (11+1)의 제곱인 144를 리턴하고, 3이라면 ‘no’을 리턴하면 됩니다.
풀이 (python)
# 풀이 1 import math def nextSqure(n): return pow(math.sqrt(n)+1, 2) if int(math.sqrt(n)) == math.sqrt(n) else 'no' # 아래는 테스트로 출력해 보기 위한 코드입니다. print("결과 : {}".format(nextSqure(121))); # 풀이 2 def nextSqure(n): for i in range(1,n//2): if i * i == n: return (i + 1) ** 2 if i * i > n: return 'no' # 풀이 3 def nextSqure(n): t = int(pow(n, 0.5)) return (t+1)**2 if t**2==n else 'no' # 참고 - 제곱근을 구하는 여러가지 방법 # n ** 5 # pow(n, 0.5) # math.sqrt(n) 다른사람 풀이
def nextSqure(n): sqrt = n ** (1/2) if sqrt % 1 == 0: return (sqrt + 1) ** 2 return 'no' def nextSqure(n): sqrt = pow(n, 0.5) return pow(sqrt + 1, 2) if sqrt == int(sqrt) else 'no' # 아래는 테스트로 출력해 보기 위한 코드입니다. print("결과 : {}".format(nextSqure(121))); 배운점
- 파이썬 삼항 연산자 (conditional expressions)을 활용하니 짧게 코드를 짤 수 있었다.
- 제곱근을 구하려면
import math를 통해서 모듈을 임포트하고math.sqrt()함수를 통해서 제곱근을 구할 수 있다. - int() 를 통해서 주어진 인수를 정수로 변환할 수 있다.
pow(4,2)과4 ** 2(=16) 을 통해서 제곱 연산을 할 수 있다.- 반대로
pow(4,0.5)과4 ** 1/2(=2) 을 통해서 제곱근을 구할 수 있다.