[백준 BOJ / C++] 1541번: 잃어버린 괄호

 

문제링크: www.acmicpc.net/problem/1541

1541번: 잃어버린 괄호

 

풀이

 

식을 마이너스 연산자가 나오기 전과 후로 나눠서 생각할 수 있다. 한 번 마이너스 연산자가 나온 후에는 뒤에 나오는 숫자들은 모두 마이너스로 계산할 수 있다. 

 

예를 들어 1+1+1-1+1-1+1과 같은 식이 있을 때 최솟값을 가지도록 괄호를 치면 1+1+1-(1+1)-(1+1)이 된다. 한 번 -가 나오면 뒤에 나오는 +는 앞의 연산과 괄호로 합쳐서 뺄셈을 해주는 것이다. 

 

따라서 bool 자료형인 minus라는 변수를 만들어서 이 값이 false면 숫자를 더하고, true면 빼준다. 문자열의 마지막에도 연산을 진행해야 하기 때문에 for문이 i <= s.size()까지 실행된다. 

 

 

 

 

소스코드

 

//1541번: 잃어버린 괄호
#include <iostream>
#include <string>
 
using namespace std;
 
int solution(string s) {
    int answer = 0;
    bool minus = false;
    string temp = "";
 
    for(int i = 0; i <= s.size(); i++) {
        if(s[i] == '+' || s[i] == '-' || s[i] == '\0') {
            if(minus)
                answer -= stoi(temp);
            else
                answer += stoi(temp);
            temp = "";
            if(s[i] == '-')
                minus = true;
        }
        else {
            temp += s[i];
        }
    }
 
    return answer;
}
 
int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
 
    string s;
    cin >> s;
    cout << solution(s);
 
    return 0;
}

 

 

 

공부한 것을 정리한 내용입니다. 수정할 부분이 있다면 알려주시면 감사하겠습니다 :)