백준 C++ 4949번) 균형잡힌 세상

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

4949번: 균형잡힌 세상

 

Stack를 활용하여 문제를 해결하였습니다. 단순이 '(' 나 '[' 일때는 스택에 넣어주고, ')', ']'를 만나면 스택에 있는 것과 비교해서 있으면 pop 후 넘어가고 없으면 break 시켜서 판별하였습니다.

 

#include<iostream>
#include<string>
#include<stack>

using namespace std;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(nullptr);
	string s;
	while (1) {
		stack<char> bal;
		getline(cin, s);
		if (s == ".")
			break;
		for (auto c : s) {
			if (c == '[')
				bal.push(c);
			else if (c == '(')
				bal.push(c);
			else if (c == ']') {
				if (bal.empty()) {
					bal.push(c);
					break;
				}
				if (bal.top() == '[')
					bal.pop();
				else
					break;
			}
			else if (c == ')') {
				if (bal.empty()) {
					bal.push(c);
					break;
				}
				if (bal.top() == '(')
					bal.pop();
				else
					break;
			}
		}
		if (bal.empty())
			cout << "yes" << "\n";
		else
			cout << "no" << "\n";
	}
}