[백준] 소수 단어 (2153)(kotlin)

문제 설명

백준 2153번 문제 링크

입력 및 출력

» 입력

첫째 줄에 단어가 주어진다. 단어의 길이는 20자 이하이다. 주어지는 단어는 알파벳 소문자와 대문자만으로 이루어져 있다.

» 출력

아래의 예제와 같은 형식으로 출력을 한다. 소수 단어인 경우에는 It is a prime word.를, 아닌 경우에는 It is not a prime word.를 출력한다.

예제 입출력(테스트케이스)

입력	출력
UFRN	It is a prime word.

문제 풀이1

fun main(args: Array<String>) = with(System.`in`.bufferedReader()) {
    var sum = 0

    readLine().forEach {
        sum += if (it in 'a'..'z') it - 'a' + 1 else it - 'A' + 27
    }

    println(primeCheck(sum))
}

fun primeCheck(pn: Int): String {
    for (i in 2 until pn)
        if (pn % i == 0) {
            return "It is not a prime word."
        }
    return "It is a prime word."
}