[BOJ 백준] 4949번 균형잡힌 세상 / C++

문제

---

문제를 보시려면 링크를 클릭해주세요.

풀이

---

STL stack을 사용하여 풀었습니다.

1. 여는 괄호가 나오면 스택에 추가합니다.
2. 닫는 괄호가 나왔을 경우,
   • 스택이 비어있으면 올바른 괄호쌍이 아니므로 반복문을 탈출합니다.
   • 스택의 top이 올바른 괄호 쌍이면 pop 해주고, 올바른 괄호 쌍이 아니면 반복문을 탈출합니다.
3. 위의 과정이 끝난 뒤, 올바른 괄호 쌍이면서, 스택이 비어있는 경우를 균형이 잡혀있다고 볼 수 있습니다. (코드 상에서 flag = true && S.empty() == 1)

소스 코드

---

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

using namespace std;

int main(int argc, char**argv) {

	ios::sync_with_stdio(0);
	cin.tie(0);

	while(true){
		stack<char> S;
		string str;
		
		getline(cin, str);
		
		if(str==".")
			break;
	
		bool flag = true;
			
		for(int i=0; i<str.size(); i++){
			
			if(str[i] == '[' || str[i] == '('){
				S.push(str[i]);
			}
			
			else if(str[i] == ']'){
				if(S.empty()){
					flag = false;
					break;
				}
				
				else if(S.top() == '['){
					S.pop();
				}
				
				else if(S.top() == '('){
					flag = false;
					break;
				}
			}
			
			else if(str[i] == ')'){
				if(S.empty()){
					flag = false;
					break;
				}
				
				else if(S.top() == '('){
					S.pop();
				}
				
				else if(S.top() == '['){
					flag = false;
					break;
				}
			}
			
		}
		
		if(S.empty() && flag)
			cout << "yes\n" ;
			
		else
			cout << "no\n" ;
	
	}

	return 0;
}