[백준] 연길이의 이상형 (20540)(kotlin)

문제 설명

백준 20540번 문제 링크

입력 및 출력

» 입력

연길이의 MBTI 4글자가 대문자로 주어진다.

» 출력

연길이의 이상형에 해당하는 MBTI 4글자를 대문자로 출력한다.

예제 입출력

입력	출력
ESTJ	INFP
INFP	ESTJ

문제 힌트

MBTI는 Myers-Briggs Type Indicator의 줄임말이다.

문제 풀이1

import java.util.Scanner

fun main(args: Array<String>) = with(Scanner(System.`in`)){
    println(
        readLine()!!
            .map { invert(it) }
            .joinToString("")

    )
}

fun invert(char: Char): String = when (char) {
    'I' -> "E"
    'S' -> "N"
    'T' -> "F"
    'J' -> "P"
    'E' -> "I"
    'N' -> "S"
    'F' -> "T"
    else -> "J"
}