[문자열 / Kotlin] 2019 KAKAO BLIND RECRUITMENT - 오픈채팅방

문제

풀이 언어

Kotlin

코드

class Solution {
    fun solution(record: Array<String>): Array<String> {
        val history = arrayListOf<Pair<String, String>>()
        val name = mutableMapOf<String, String>()
        val result = arrayListOf<String>()

        for (e in record.iterator()) {
            val act = e.split(" ")
            when (act[0]) {
                "Enter" -> {
                    name[act[1]] = act[2]
                    history.add(Pair(act[0], act[1]))
                }
                "Leave" -> {
                    history.add(Pair(act[0], act[1]))
                }
                "Change" -> {
                    name[act[1]] = act[2]
                }
            }
        }

        history.map {
            when (it.first) {
                "Enter" -> result.add("${name[it.second]}님이 들어왔습니다.")
                else -> result.add("${name[it.second]}님이 나갔습니다.")
            }
        }

        return result.toTypedArray()
    }
}

풀이 방법

문제를 효율적으로 해결하기 위해서는 Map 을 사용해야한다. Map은 Key - Value 의 형태로 저장하는 Collection 의 한 종류인데 이를 활용하여 아이디 - 닉네임 꼴로 저장한다. Change 가 입력된 경우 아이디 - 닉네임 을 갱신해준다. 이후 Map 을 참조하면서 채팅방 로그를 전달하면 된다.

결과