[백준 1541] 잃어버린 괄호(Python, Java)

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

1541번: 잃어버린 괄호

문제

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

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

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

입력

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

출력

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

예제 입력 1

55-50+40

예제 출력 1

-35

풀이 설명

1. 무조건 + 연산부터 죄다 해놓고 남은 - 연산들을 수행하는게 답이다.

2. 입력받은 문자열을 숫자 리스트와 연산자 리스트로 나누어 저장한다.

3. 연산자 리스트를 탐색하면 덧셈이면 해당 위치의 숫자들을 더하고, 연산자와 숫자들을 제거한 후 더한 숫자를 그 위치에 삽입한다.

4. 남은 - 연산들을 수행한다.

Python3 코드

s = input()
ops = []
nums = []
num = ''
for c in s:
    if c == '+' or c == '-':
        nums.append(int(num))
        num = ''
        ops.append(c)
    else:
        num += c
nums.append(int(num))

idx = 0
while(idx < len(ops)):
    if ops[idx] == '+':
        a = nums[idx]
        b = nums[idx+1]
        del(ops[idx])
        del(nums[idx])
        del(nums[idx])
        nums.insert(idx, a+b)
    else:
        idx += 1
        
answer = nums[0]
for i in range(1, len(nums)):
    answer -= nums[i]
print(answer)

Java 코드

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

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String s = br.readLine();
		List<Character> ops = new ArrayList<>();
		List<Integer> nums = new ArrayList<>();
		
		String num = "";
		for(int i=0; i<s.length(); i++) {
			char c = s.charAt(i);
			if(c == '+' || c == '-') {
				nums.add(Integer.parseInt(num));
				num = "";
				ops.add(c);
			}else {
				num += Character.toString(c);
			}
		}
		nums.add(Integer.parseInt(num));
		
		int idx = 0;
		while(idx < ops.size()) {
			if(ops.get(idx) == '+') {
				int a = nums.get(idx);
				int b = nums.get(idx+1);
				ops.remove(idx);
				nums.remove(idx);
				nums.remove(idx);
				nums.add(idx, a+b);
			}
			else idx++;
		}
		
		int answer = nums.get(0);
		for(int i=1; i<nums.size(); i++)
			answer -= nums.get(i);
		
		System.out.println(answer);
	}
}