본문 바로가기

개발 노트/알고리즘[Kotlin]

프로그래머스 알고리즘 문제[정수 제곱근 판별]

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

 

 

#제한사항

n은 1이상, 50000000000000 이하인 양의 정수입니다.

 

 

#입출력 예

ex) 121은 양의 정수 11의 제곱이므로, (11+1)를 제곱한 144를 리턴합니다.

ex) 3은 양의 정수의 제곱이 아니므로, -1을 리턴합니다.

 

 

# 풀이

import kotlin.math.*
class Solution {
    fun solution(n: Long): Long {
        val x = sqrt(n.toDouble()).toLong()
        
        return if(n == x*x) (x+1)*(x+1)
        else -1L
        
    }
}

 

제곱근을 구할때는 sqrt연산을 사용해서 간단하게 할 수 있다

근데 sqrt는 반환형이 기본적으로 Double형이기 때문에, 정수형으로 구하기위해서는 toDouble을 사용해 형변환 후 사용해줘야한다. 그런데 원래 Long타입이기 때문에 마지막에 Long타입으로 형변환을 시켜서 되돌려준다

그리고 Long타입이기때문에 숫자옆에 L을 붙여줘야 오류가 생기지 않는다

 

 

[제곱,제곱근,세제곱근 연산 정리]

 

sqrt : 제곱근!!

println(sqrt(4.0))  //2.0출력

 

pow : 제곱수

println(2.0.pow(3)) // 8.0출력

 

cbrt : 세제곱근 

println(cbrt(8.0))	// 2.0출력

 

hypot : 두 수 각각의 제곱수를 합한 수의 제곱근 

println("${hypot(3.0, 4.0)}")	// 5.0출력  , (3제곱+4제곱)의 제곱근

 

 

-> 위 연산들의 기본반환형은 모두 Double형이다

-> 위 연산들을 사용하기위해서는 기본적으로 인포트를 해줘야한다 ( import kotlin.math.*)