[백준, BOJ 1541] 잃어버린 괄호(java)

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

1541번: 잃어버린 괄호

---

문제

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

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

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

입력

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

출력

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

728x90

 

예제 입력 1

55-50+40

예제 출력 1

-35

 

예제 입력 2

10+20+30+40

예제 출력 2

100

 

예제 입력 3

00009-00009

예제 출력 3

0

---

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;

public class Main {

	public static void main(String[] args) throws Exception {

		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		char[] calculation = br.readLine().toCharArray();
		
		ArrayList<Integer> num = new ArrayList<>();
		ArrayList<Character> operator = new ArrayList<>();
		
		// 숫자는 num에 연산자는 operator에 추가
		int nIdx = 0;
		for (int i = 0; i < calculation.length; i++) {
			if(calculation[i] == '+' || calculation[i] == '-') {
				int tmp = 0;
				for (int j = nIdx; j < i; j++) {
					tmp = tmp * 10 + (calculation[j] - '0');
				}
				num.add(tmp);
				operator.add(calculation[i]);
				nIdx = i + 1;
			}
		}
		// 마지막 숫자 추가
		int tmp = 0;
		for (int i = nIdx; i < calculation.length; i++) {
			tmp = tmp * 10 + (calculation[i] - '0');
		}
		num.add(tmp);
		
		// - 연산이 나오면 다음 - 연산이 나올 때까지 사이에 있는 + 연산을 모두 먼저 계산
		for (int i = 0; i < operator.size();) {
			if(operator.get(i) == '-') {
				i++;
				while(i < operator.size() && operator.get(i) == '+') {
					num.set(i, num.get(i) + num.get(i + 1));
					num.remove(i + 1);
					operator.remove(i);
				}
			}
			else i++;
		}
		
		// 남은 연산 마저 계산
		int result = num.get(0);
		for (int i = 0; i < operator.size(); i++) {
			if(operator.get(i) == '+') result += num.get(i + 1);
			else result -= num.get(i + 1);
		}
		
		// 출력
		System.out.println(result);
	}

}