프로그래머스 - 다항식 더하기 (Kotlin)

프로그래머스

 

이 문제, 가볍게 생각했다가 생각보다 생각해야할 것이 조금 있었다. 내가 생각한 문제 풀이의 핵심은 x계수 와 숫자를 어떻게 합산할 것인가 이다.

• x의 계수는 두자리 이상이 될 수 있음을 주의해야한다.(이것 때문에 시간을 한참 사용했다..)
• 반환하는 문자열의 조건에 주의한다.(계수가 1이거나, 0 인 경우 어떻게 처리할 것인가)

나같은 경우, x의 계수가 두자리 이상이 올 수 있음을 미처 생각하지 못해 다른 풀이를 참고했다. 그리고 반환하는 문자열의 경우, 계수가 0일 경우, 1일경우, 숫자가 0일경우를 잘 다뤄야 한다.

 

자꾸만 기본적인 조건을 다루는 문제에 대해 실수를 하는게 부끄럽다.. 아무리 난이도가 낮은 문제일지라도, 좀 더 진지하게 문제를 풀이해야겠다.

class Solution {
    fun solution(polynomial: String): String {
        var xCount = 0
        var num = 0

        val terms = polynomial.split(" + ")

        terms.forEach { term ->
            if (term.contains("x")) {
                xCount += if (term == "x") 1 else term.replace("x","").toString().toInt()
            } else {
                num += term.toInt()
            }
        }

        return (if (xCount != 0) if (xCount > 1) "${xCount}x" else "x" else "") + if (num != 0) (if (xCount != 0) " + " else "") + num else if (xCount == 0) "0" else ""
        
    }
}

 

참고블로그

[Algorithm] 프로그래머스 - 다항식 더하기 (Kotlin)