[BOJ] 1541. 잃어버린 괄호 ✔

1541. 잃어버린 괄호

1. 문제

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

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

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

2. 입력

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

3. 출력

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

4. 풀이

Python

fs = input().split('-') # '-'를 기준으로 슬라이싱

result = 0
for i, f in enumerate(fs):
    if i == 0:
        result += sum(map(int, f.split('+')))
        # 첫 '+' 수식값만 더해준 뒤
    else:
        result -= sum(map(int, f.split('+')))
        # 나머지 '+' 수식값은 모두 뺀다

print(result)

C++

using namespace std;
#include <iostream>
#include <string>

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    
    string str;
    cin >> str;
    
    // '-'가 나온 이후의 숫자는 모두 빼면 되기 때문에 체크할 수 있는 bool 변수 선언
    bool isMinus(false);
    int num(0), result(0);
    for (int i = 0; i <= str.length(); ++i)
    {
        if (str[i] == '+' || str[i] == '-' || i == str.length())
        {
            if (isMinus) result -= num;
            else result += num;
            
            // 부호가 나왔을 때 num 초기화 및 '-'인지 체크
            num = 0;
            if (str[i] == '-') isMinus = true;
        }
        else num = num * 10 + (str[i] - '0');
    }
    
    cout << result;
}