[4949] 균형잡힌 세상, c++

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

4949번: 균형잡힌 세상

한줄 후기 : 오랜만에 코딩하려니 힘들다..

 

 

 

괄호와 매우 비슷한 문제, 다만 '(' 이 괄호와 '[' 이 괄호의 처리를 다르게 해줘야 한다.

처음에 같이 했다가 잘못나와서 다시 보니 스택에서 pop 해줄때 top 원소가 '(' 인지 '['인지 확인해야 한다.

그리고 각 문자열을 돌릴때마다 스택을 empty상태로 만들어줘야한다.

 

#include <iostream>
#include <stack>
#include <string.h>
using namespace std;
stack<char> s;
char str[101];
string check(char str[]);
int main(){
	while(1){
		cin.getline(str,101,'\n');
		if(str[0] == '.')	break;
		else{
			cout << check(str) << endl;
		}
		while(!s.empty())	s.pop();
	}
	return 0;
}
string check(char str[]){
	char ch;
	for(int i=0; i<strlen(str); i++){
		ch = str[i];
		if(ch == '(' || ch =='[')	s.push(ch);
		else if(ch == ')'){
			if(s.empty())	return "no";
			if(s.top()=='(')	s.pop();
			else	return "no";
		}
		else if(ch == ']'){
			if(s.empty())	return "no";
			if(s.top()=='[')	s.pop();
			else	return "no";
		}
	}
	if(s.empty())	return "yes";
	else	return "no";
}