1541번: 잃어버린 괄호

문제 출처 : https://www.acmicpc.net/problem/1541

 

언어 : Kotlin

 

문제 설명 :

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

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

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

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

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 128MB

 

입출력 예 :

입력	출력
55-50+40	-35
10+20+30+40	100
00009-00009	0

풀이 : 

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val str = br.readLine().split("-")
    val arr = mutableListOf<Int>()
    for (s in str) {
        val temp = s.split("+")
        arr.add(temp.sumOf { it.toInt() })
    }

    var result = arr[0]
    (1 until arr.size).forEach { result -= arr[it] }

    bw.write("$result")
    bw.flush()
    bw.close()
}

 

처음에는 복잡하게 -만 나올때, +만 나올때 섞여서 나올때를 다 나눠서 계산하려 했으나 생각해보니 곱셈 나눗셈이 있는 것이 아니라서 단순히 덧셈부터 해준 뒤 빼주면 최솟값이 나오는 것이었다.