백준 4949 균형잡힌 세상

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

문제와 채점은 위 사이트에서 확인 하실 수 있습니다.

 

 

이 문제는 각각 대괄호'[]' 와 소괄호() 가 알맞게 짝지어 져있는지 확인 하는 문제였습니다.

 

따라서 '(' 이나 '[' 가 나오면 벡터에 push를 해주고 ')' 이나 ']' 가 나오는 순간 벡터의 back 부분에 그에 짝이 맞는 괄호가 있는지 확인 한 후 pop을 해주었습니다. 닫히는 괄호가 나오는 순간 그 이전에 열린괄호가 짝이 맞지 않다면 괄호가 균형잡히게 들어 있지 않은 것입니다.

 

저는 비어있는 조건 검사하기가 귀찮아서 초기에 'a'를 그냥 넣어주었습니다.

 

다음은 코드입니다.

 

#include <iostream>
#include <vector>
#include <string>

using namespace std;

int main(void) {
	ios_base::sync_with_stdio(false); cin.tie(NULL);
	string str1;
	vector<char> v;
	bool flag;

	while(1) {
		flag = true;
		getline(cin, str1);
		
		if(v.empty()) //비어 있으면 문자'a'삽입 
			v.push_back('a');
		
		if(str1.length() == 1 && str1[0] == '.') //.이 나오면 입력 종료 
			break;

		for(int i = 0; i < str1.length(); i++) {
			if(str1[i] == '(' || str1[i] == '[') //'('이나 '["가 나오면 push 
				v.push_back(str1[i]);
			
			if(str1[i] == ')') { //')가 나왔을때 '('가 있으면 pop 그렇지 않으면 break 
				if(v.back() == '(') 
					v.pop_back();
				else {
					flag = false;
					break;
				}
			}
			else if(str1[i] == ']') { //']'가 나왔을때 '['가 있으면 pop 그렇지 않으면 break 
				if(v.back() == '[') 
					v.pop_back();
				else {
					flag = false;
					break;
				}
			}					
		}
		
		if(flag && v.size() == 1) //flag가 true이고 벡터에 'a'말고 문자가 없다면 yes 
			cout<<"yes"<<'\n';
		else
			cout<<"no"<<'\n';
		
		v.clear();
	}
	return 0;

}