백준 1541: 잃어버린 괄호 (그리디 알고리즘) -Java

실버2

https://www.acmicpc.net/problem/1541

 

문제

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

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

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

 

입력

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

 

출력

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

 

예제 입력

55-50+40

 

예제 출력

-35

 

자바 코드

package practice;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    static int[][][] map = new int[101][101][101];

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        
        // 1. 뺄셈을 기준으로 토큰화
        StringTokenizer st = new StringTokenizer(br.readLine(), "-");

        int result = 0;
        int cnt = 0;
        
        // 2. 토큰 내부는 모두 덧셈이므로 전부 합한다.
        while(st.hasMoreTokens()) {
            int minusTotal = 0;

            String stNextToken = st.nextToken();
            StringTokenizer st2 = new StringTokenizer(stNextToken, "+");
            
            while(st2.hasMoreTokens()) {
                minusTotal += Integer.parseInt(st2.nextToken());
            } 
            
            if (cnt == 0) {
                // 첫번째 토큰만 더한다 
                result += minusTotal;
            } else {
                result -= minusTotal;
            }
            cnt++;
        }
        
        System.out.println(result);

    }
}

 

 

PS

"괄호를 적절히 친다" 라는 말 때문에 조금 헷갈렸다. 괄호를 하나만 치는건지 여러개 칠 수 있는건지 모르겠었고, 예제 입력 중에도 괄호를 하나 만 치는 입력값 밖에 없었다. 조금 헤매다가 그냥 덧셈이 있는 항들을 전부 괄호로 치고 (토큰화), 첫번째 수에서 전부 빼면 된다는 사실을 깨닫고 다시 코딩했다. 이 방법이 그리디 알고리즘을 알맞게 적용한 것인지는 잘 모르겠다..