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

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

4949번: 균형잡힌 세상

소스 코드 : 

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

int main()
{
	stack<string> stack;
	string str;
	getline(cin, str);
	int size = str.length();
	while (1)
	{
		if (str == ".")
		{
			break;
		}
		
		int num = 0;

		for(int i=0;i<size;i++)
		{
			if (str[i] == '(')
			{
				string element = "(";
				stack.push(element);
				num++;
			}

			else if (str[i] == '[')
			{
				string element = "[";
				stack.push(element);
				num++;
			}

			else if (str[i] == ')')
			{
				if (stack.size() != 0)
				{
					string check = stack.top();
					if (check == "(")
						stack.pop();
				}
				num--;
			}

			else if (str[i] == ']')
			{
				if (stack.size() != 0)
				{
					string check = stack.top();
					if (check == "[")
						stack.pop();
				}
				num--;
			}
		}

		if (stack.empty() && num==0)
			cout << "yes" << '\n';
		else
			cout << "no" << '\n';

		while(!stack.empty())
			stack.pop();
		getline(cin, str);
		size = str.length();
	}
}

마무리 : 스택을 이용해서 '(', '['이 오면 스택에 push를 했다가 ')', ']' 나올 때 pop을 해서 문장이 끝날 때 균형이 잡힌 문자열이면 'yes'를 출력하고 아니면 'no'를 출력하는 문제이다. 스택을 공부할 때 많이 푸는 문제 중 하나이다.