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

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

1541번: 잃어버린 괄호

 

 

---

 

 

문제

---

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

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

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

 

입력

---

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

 

출력

---

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

 

예제 입력 1

---

55-50+40

예제 출력 1

---

-35

 

예제 입력 2

---

10+20+30+40

 

예제 출력 2

---

100

 

코드 

---

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

int main() {
	string a, b = "";
	int n1 = 0, n2 = 0, res = 0;
	bool flag = false;
	cin >> a;

	for (size_t i = 0; i <= a.length(); i++) {
		if (a[i] == '+' || a[i] == '-' || a[i] == '\0') {
			if (flag) res -= stoi(b);
			else res += stoi(b);
			if (a[i] == '-') flag = true;
			b = "";
			continue;
		}
		b += a[i];
	}

	cout << res;
	return 0;
}

 

 

 

---

 

 

처음에 문제를 잘 이해하지 못해서 좀 헤맸다. 괄호는 원하는 위치에 삽입하는 식으로 쳐지는 것이며, 여러 번 칠 수 있다.

이 문제를 풀기 위한 아이디어는 매우 단순하다. 

입력의 첫 번째 문자는 무조건 숫자이므로, 양수일 것이다. 식에서 가장 작은 수를 만들기 위해서는 첫 번째 수부터 - 기호가 나타나기 전까지는 계속 더하다가, 최초로 - 기호가 나타나는 곳부터는 무조건 뺄샘만 하도록 괄호를 치면 된다.

예시를 들어 아래와 같은 방식으로 괄호를 치면 된다.

1 + 2  + 3 - 4 + 5 + 6 - 7 + 8    ->    1 + 2 + 3 - (4 + 5 + 6) - (7 + 8) = 24

1 + 2 + 3 - 4 - 5    -> 1 + 2 + 3 - (4) - (5) = -3

따라서 어떤 식에서 가장 작은 값은 - 기호가 처음 나오는 순간부터 모든 수들을 뺄셈 연산만 한 값인 것이다.

이를 알고리즘으로 구현하기 위해 문자를 하나씩 읽고, + 연산자, - 연산자, 문자열 끝을 나타내는 \0 를 만났을 때 문자로 이루어진 수를 int형 숫자로 바꾸어 각각 필요한 연산을 해주었다. 연산은 flag에 의해 결정되도록 구현했다. for문을 돌다가 - 기호를 만나게 되면 flag 값을 확인한다.  flag가 false일 때는 + 연산을 하고, true 일 때는 - 연산을 한다. 입력의 첫 번째 문자는 숫자이기 때문에 flag의 초기값은 false 여야 한다.