[백준] 1541번 잃어버린 괄호 - 자바

1541번: 잃어버린 괄호

문제

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

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

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

입력

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

출력

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

예제 입출력

풀이

이 식의 값이 최소가 되게 하려면, 최대한 많은 숫자에 (-)부호를 붙이면 된다.

• (-) 부호가 처음으로 나오기 전까지의 숫자들은 괄호를 어떻게 넣어도 (-)를 붙일 수 없다.
• (-) 부호가 한번이라도 나오면 그 후 나오는 숫자들은 모두 (-)를 붙일 수 있다.

예를 들어, 주어진 식이 3 + 4 - 2 + 3 + 3 - 5 - 2 + 9 라면, 

3 + 4 - (2 + 3 + 3) - 5 - (2 + 9) = 7 - 8 - 5 - 11 = -17

이렇게 처음 (-)가 나오는 지점 이후로 모든 숫자에 (-)를 붙일 수 있다.

 

위 로직을 코드로 구현

1. 주어진 식을 "-" 기준으로 분리

2. 분리된 식을 각각 계산하여 리스트에 넣기

3. 리스트의 첫번째 요소는 (+)부호를 붙여 더하고, 나머지 요소는 (-)부호를 붙여 더하기

 

코드

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String expression = br.readLine();

        List<Integer> numbers = new ArrayList<>();
        for (String expr : expression.split("-")) {
            String[] nums = expr.split("\\+");
            int number = 0;
            for (String num : nums) {
                number += Integer.parseInt(num);
            }
            numbers.add(number);
        }

        int answer = numbers.get(0);
        for (int i = 1; i < numbers.size(); i++) {
            answer -= numbers.get(i);
        }

        System.out.println(answer);
    }
}