[Algorithm] 프로그래머스 - 조건 문자열 (Kotlin)

 

받은 문자열이 무엇이냐에 따라 n, m에 수행하는 연산이 달라지고 리턴하는 숫자가 달라지기 때문에 when 절을 쓰면 될 것이다.

 

class Solution {
    fun solution(ineq: String, eq: String, n: Int, m: Int): Int = when {
        ineq == "<" && eq == "=" -> if (n <= m) 1 else 0
        ineq == ">" && eq == "=" -> if (n >= m) 1 else 0
        ineq == "<" && eq == "!" -> if (n < m) 1 else 0
        ineq == ">" && eq == "!" -> if (n > m) 1 else 0
        else -> {0}
    }
}

 

이전에 when을 써서 풀었을 때와 마찬가지로, else 브랜치가 거슬린다. 없애려면 if를 쓰거나 when과 중괄호 사이에 소괄호로 조건을 넣으면 된다. 이대로도 통과됐고 귀찮아서 그렇게 하지 않았다.