[Baekjoon/C++] 1541번 - 잃어버린 괄호

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

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

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

 

입력

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

 

출력

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

 

---

예제 입력	예제 출력
55-50+40	-35
10+20+30+40	100
00009-00009	0

---

풀이

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

vector<int> arr; // 숫자
vector<bool> op; // 기호

int small();

int small() {
	int result = arr[0];
	int idx = 0;

	for (int i = 1; i < arr.size(); i++) {
		if (!op[idx])
			result -= arr[i];
		else {
			result += arr[i];
			idx++;
		}
	}

	return result;
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	string s, a = "";
	cin >> s;

	// 식의 숫자와 기호를 분리
	for (int i = 0; i <= s.length(); i++) {
		if (s[i] == '-' || s[i] == '+' || i == s.length()) {
			arr.push_back(stoi(a));

			if (s[i] == '-')
				op.push_back(false);
			else
				op.push_back(true);

			a = "";
		}
		else a += s[i];
	}

	cout << small() << '\n';

	return 0;
}

 

	// 식의 숫자와 기호를 분리
	for (int i = 0; i <= s.length(); i++) {
		if (s[i] == '-' || s[i] == '+' || i == s.length()) {
			arr.push_back(stoi(a));

			if (s[i] == '-')
				op.push_back(false);
			else
				op.push_back(true);

			a = "";
		}
		else a += s[i];
	}

s에 입력 받은 식의 숫자와 기호를 분리하는 코드이다. arr에 stoi를 통해 int로 변환한 수를 op에 +와 -의 여부를 저장하도록 했다.

 

int small() {
	int result = arr[0];
	int idx = 0;

	for (int i = 1; i < arr.size(); i++) {
		if (!op[idx])
			result -= arr[i];
		else {
			result += arr[i];
			idx++;
		}
	}

	return result;
}

입력 받은 식의 최솟값을 구하는 함수이다. result에 arr의 숫자들을 모두 더하다가 op[idx]가 -라면 그 뒤의 숫자들은 모두 빼기 연산을 하도록 했다.