[Swift] Programmers_Lv1 정수 제곱근 판별

제 문제 풀이는 최선의 풀이가 아닐 수 있습니다. 문제를 푸는 방법은 다양합니다!

 

문제설명

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

 

제한 조건

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

 

입출력 예

n	return
121	144
3	-1

문제 분석

1. 제곱근을 판별하기 위해 sqrt() 함수를 사용.

2. 함수를 사용하기 위해 Foundation을 import 해야 함.

3. 양의 정수를 가지는 값이라면 + 1을 하여 제곱하고, 소수점을 가지는 값이 나온다면 -1을 리턴함.

 

Solution.swift

import Foundation

func solution(_ n:Int64) -> Int64 {
    
    // 결과값을 담을 변수 선언
    var result: Int64 = 0
    
    // sqrt()를 사용해 n의 제곱근을 구함
    let number = sqrt(Double(n))
    
    // Double형의 number에 소숫점값이 있다면
    // Int64로 number를 감쌋을때 소숫점값이 다 떨어져 나가게 됨
    // 값 비교를 위해 다시 Double형으로 선언함
    if number == Double(Int64(number)) {
        result = Int64((number + 1) * (number + 1))
    } else {
       result = -1 
    }
    
    return result
    
}

 

프로그래머스 인기 풀이

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