[Algorithm] 프로그래머스 - 왼쪽 오른쪽 (Kotlin)

 

class Solution {
    fun solution(str_list: Array<String>): Array<String> {
        val indexL = str_list.indexOf("l")
        val indexR = str_list.indexOf("r")

        return when {
            indexL != -1 && (indexR == -1 || indexL < indexR) -> str_list.sliceArray(0 until indexL)
            indexR != -1 -> str_list.sliceArray((indexR + 1) until str_list.size)
            else -> arrayOf()
        }
    }
}

 

l과 r의 인덱스를 가져와서, 조건에 맞게 l 앞의 원소들과 r 뒤의 원소들을 배열에 담아 리턴한다.

문제에는 리스트라고 써 있는데 리턴타입은 문자열 배열이니 헷갈리지 말자.