[프로그래머스][Lv. 1] 정수의 제곱근 판별 Swift 풀이

https://programmers.co.kr/learn/courses/30/lessons/12934

코딩테스트 연습 - 정수 제곱근 판별

풀이

 

간단한 문제이다.

입력으로 주어진 Int64 자료형 정수 n의 제곱근을 sqrt() 로 구한 다음, 이를 Int로 형변환한 값과 정확히 같은 지 확인해보면 정수의 제곱근 여부를 간단히 확인할 수 있다.

정수의 제곱 여부를 확인하여 맞다면 그 정수+1의 제곱을 반환하고, 그렇지 않을 경우 -1을 반환하면 된다.

 

끝!

import Foundation
 
func solution(_ n:Int64) -> Int64 {
    var tmp: Double = sqrt(Double(n))
    return tmp == Double(Int(tmp)) ? Int64(pow(Double(tmp + 1), 2.0)) : -1 
}