[Swift] 프로그래머스 입문 문제, 제곱수 판별하기

문제

 

풀이 방법

1. 함수 인자를 sqrt로 제곱근을 return한다 ex) 4가 함수인자면 sqrt는 2가 나옴
2. 10을 곱하고 반올림을 한뒤 10을 나눠준다
3. if문에 break되면 제곱근이고
4. else문에 break되면 제곱근이 아니다

 

코드

import Foundation

func solution(_ n:Int) -> Int {
    var roundedRating = round(sqrt(Double(n)) * 10) / 10
    
    if roundedRating.truncatingRemainder(dividingBy: 1) == 0 { // 제곱근 Case
        return 1
    } else { // 제곱근이 아닌 Case
        return 2
    }
}