[백준] 1541번 잃어버린 괄호 해답(Kotlin, 코틀린)

문제

세준이는 양수와 +, -, 그리고 괄호를 가지고 식을 만들었다. 그리고 나서 세준이는 괄호를 모두 지웠다.

그리고 나서 세준이는 괄호를 적절히 쳐서 이 식의 값을 최소로 만들려고 한다.

괄호를 적절히 쳐서 이 식의 값을 최소로 만드는 프로그램을 작성하시오.

입력

첫째 줄에 식이 주어진다. 식은 ‘0’~‘9’, ‘+’, 그리고 ‘-’만으로 이루어져 있고, 가장 처음과 마지막 문자는 숫자이다. 그리고 연속해서 두 개 이상의 연산자가 나타나지 않고, 5자리보다 많이 연속되는 숫자는 없다. 수는 0으로 시작할 수 있다. 입력으로 주어지는 식의 길이는 50보다 작거나 같다.

출력

첫째 줄에 정답을 출력한다.

---

 

알고리즘

1. +와 -의 연산순서를 통해서 가장 큰 값을 구하기 위해서는 +를 먼저 연산 즉 ->  +가 포함된 수열에 잃어버린 괄호를 씌워주는 것이다.
2. 먼저 첫 번째 숫자의 경우 무조건 +의 값으로 판단되기에 따로 first의 변수를 생성한다.
3. +가 포함 된 식이라면 계산에 우선순위에 둬야하기에 plus 변수에 따로 연산한다.
4. -가 포함 된 식이라면 계산에서 뺄셈을 진행해야하기에 minus 변수에 따로 연산한다.

 

정답 코드

fun main() {
    var first = 0
    var plus = 0
    var minus = 0
    readln().split("-").forEachIndexed { index, s ->
        if (index == 0) {
            first += s.split("+").sumOf { it.toInt() }
        } else {
            if (s.contains("+")) {
                plus += s.split("+").sumOf { it.toInt() }
            } else {
                minus += s.toInt()
            }
        }
    }
    println(-minus - plus + first)
}

난이도: 하

 

헷갈렸던 부분 X: 실버 2이길래 생각보다 쫄면서 문제를 풀었는데 막상 대단한 함정 같은 건 없었다. 아무래도 제출 순서대로 하다보니 문제가 쉬운게 걸린 것 같다.

 

한번에 통과~~~!!!

 

 

다른사람의 풀이

1줄로 완성하신 분도 있었다... 리스펙 accumulate(데이터를 모으는) 함수인 reduce()와 fold()가 있는데 그 중 reduce를 사용하여 total값을 Return한 것으로 파악하였다.