알고리즘 - 프로그래머스 정수 제곱근 판별 (Swift)

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

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

입출력 예

nreturn

121	144
3	-1

입출력 예 설명

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

입출력 예#2
3은 양의 정수의 제곱이 아니므로, -1을 리턴합니다.

---

네 제곱근을 구해서 제곱근이 있을 경우에 1식 더해서 다시 제곱을 해주어 반환해주는 내용입니다.

 

그렇게 어렵지 않지요. 기본적으로 풀어보면 아래와 같습니다.

func solution(_ n:Int64) -> Int64 {
    for index in 1...Int(n) {
        if index * index == n {
            return Int64((index+1) * (index+1))
        }
    }
    
    return -1
}

 

그런데 Swift에서는 sqrt라는 함수를 지원해주조

네 제곱근을 알아서 구해줍니다.

 

아래와 같이 제곱근을 구해주는 함수를 이용해서 변수로 선언해두고

그 내용을 3항식을 이용해서 판별해주고 반한해주면 됩니다.

func solution(_ n:Int64) -> Int64 {
    let squareRoot = Int64(sqrt(Double(n)))
    return squareRoot * squareRoot == n ? (squareRoot+1) * (squareRoot+1) : -1
}

 

 

위의 첫번째 풀이와 같은경우 성능면에서 많이 떨어지지만 내장항수는 0.1초 이내에 모두 값을 계산 할 수 있습니다.