[백준] 1541번 잃어버린 괄호 Java

문제 링크

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

문제

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

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

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

입력

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

알고리즘: 그리디

풀이

주어진 식에서 -가 나온 이후부터는 모든 + 연산자는 -로 바꿀 수 있다.
-로 바꿔서 계산해야 최소로 만들 수 있기 때문에 +는 -로 -는 그대로 해서 연산 후 출력한다.

소스 코드

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

public class Main {
    static boolean minus = false;
    static char operator = '+';
    static StringBuilder operand = new StringBuilder();
    static int ans = 0;
    public static void main(String args[]) throws IOException {
      BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
      String ex = br.readLine();
      for(int i=0; i<ex.length(); i++) {
          char n = ex.charAt(i);
          if(n == '+') {
              ans = calculate(operator, ans, Integer.parseInt(operand.toString()));
              operand = new StringBuilder();
              if(minus) operator = '-';
              else operator = '+';
          } else if(n == '-') {
              if(i != 0) ans = calculate(operator, ans, Integer.parseInt(operand.toString()));
              operand = new StringBuilder();
              if(!minus) minus = true;
              operator = '-';
          } else {
              operand.append(n);
              if(i == ex.length() - 1) {
                  ans = calculate(operator, ans, Integer.parseInt(operand.toString()));
              }
          }
      }
      System.out.println(ans);
    }
    static int calculate(char op, int a, int b) {
        if(op == '+') {
            return a + b;
        } else {
            return a - b;
        }
    }
}