[Lv. 0] 제곱수 판별하기

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/120909

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

• 어떤 자연수를 제곱했을 때 나오는 정수를 제곱수라고 합니다. 정수 n이 매개변수로 주어질 때, n이 제곱수라면 1을 아니라면 2를 return하도록 solution 함수를 완성해주세요.

제한 사항 : 

• 1 ≤ n ≤ 1,000,000

입출력 예 :

n	result
144	1
976	2

풀이 : 

import kotlin.math.roundToInt
import kotlin.math.sqrt

class Solution {
    fun solution(n: Int): Int {
        var answer: Int = 0
        
        answer = sqrt(n.toDouble()).roundToInt()

        return if (answer * answer == n) {
            1
        } else {
            2
        }
    }
}

간단하게 Math.sqrt 함수를 사용해 제곱근을 구한 뒤 다시 제곱하여 주어진 매개변수 n과 같다면 1을 반환, 아니라면 2를 반환한다.