백준4949 - 균형잡힌 세상

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

4949번: 균형잡힌 세상

어제 카카오 리크루트에 출제 되었던 괄호 문제를 풀지 못해 반성는 마음으로 괄호 관련 스택 문제를 풀었다. 주어진 괄호의 모양은 ()와 []가 있고 서로 맞는 괄호의 모양으로 끝을 닫아야 한다. 여는 괄호가 없는데 닫는 괄호가 나타났거나 괄호의 모양이 서로 다른데 닫으려 한다거나 탐색 결과 스택에 여는 괄호가 남아있다면 불균형적인 문장이다. 그 외에 상황은 균형잡힌 문장이다.

 

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

int main() {
	while (1) {
		stack<char> stk; //괄호를 담는 스택
		string s; //균형 잡힌지 알고 싶은 문자열
		bool check = true; //균형 잡힌 상태인지 나타내는 변수

		getline(cin, s); //공백을 포함한 입력

		//온점 하나만 입력시 종료
		if (s == ".") {
			break;
		}

		//문자열을 조사하여 스택에 하나씩 '('나 '['를 넣는다.
		for (unsigned int i = 0; i < s.size(); i++) {
			if (stk.size() == 0 && (s[i] == ']' || s[i] == ')')) { //문자열이 비었는데 닫는 괄호를 만나면
				check = false; //잘못된 균형
				break; //탐색 종료
			}
			if (s[i] == '(' || s[i] == '[') { //여는 괄호를 만났다면
				stk.push(s[i]); //스택에 넣는다.
			}
			else if (s[i] == ')') { //')'를 만났을 때
				if (stk.top() == '(') { //스택 맨 꼭대기에 '('가 있다면
					stk.pop(); //꼭대기를 빼준다.
				}
				else if(stk.top() == '[') { //스택 맨 꼭대기에 '['가 있다면
					check = false; //잘못된 균형
					break; //탐색 종료
				}
			}
			else if (s[i] == ']') { //']'를 만났을 때
				if (stk.top() == '[') { //스택 맨 꼭대기에 '['가 있다면
					stk.pop(); //꼭대기를 빼준다.
				}
				else if (stk.top() == '(') { //스택 맨 꼭대기에 '('가 있다면
					check = false; //잘못된 균형
					break; //탐색 종료
				}
			}
		}
		if (check == false || stk.size() != 0) { //탐색 결과가 틀렸거나 스택에 원소가 쌓여 있으면
			cout << "no" << '\n';
		}
		else { //탐색 결과가 맞고 스택에 원소가 없으면
			cout << "yes" << '\n';
		}
	}
}