BOJ 4949번 :: 균형잡힌 세상

#문제

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

4949번: 균형잡힌 세상

 

#작성 코드

#include <iostream>
#include <string>
#include <stack>
using namespace std;
 
int main(){
    while(true){
        string s;
        stack<char> st;
        getline(cin, s);
        
        // '.'만 입력받으면 종료한다. 
        if( s.length()==1 && s[0]=='.' )
            break;
            
        bool flag = true;
        // 입력받은 문자열의 길이만큼 반복. 모든 인덱스 방문                
        for(int i=0; i<s.length(); i++){
            if( s[i]=='('){    // 여는 괄호 만나면 push 
                st.push('(');
            }
            else if( s[i]=='[' ){
                st.push('[');
            }
            else if( s[i]==')'){        // 닫는괄호 )만났을때 
                if( !st.empty()&&st.top()=='(' ){    // 스택 top이 (이면 스택 pop 
                    st.pop();
                }
                else{
                    flag = false;
                    break;
                }
            }
            else if(s[i]==']'){            // 닫는괄호 ]만났을때 
                if( !st.empty() && st.top()=='['){    // 스택 top이 [이면 스택 pop 
                    st.pop();
                }
                else{
                    flag = false;
                    break;
                }
            }
        }
        
        if( flag && st.empty() )
            cout<<"yes\n";
        else
            cout<<"no\n";
    }
    return 0;
}

##

공백을 포함한 문자열을 입력받아야 하므로 getline(cin, str)을 사용했다.

만약 입력받은 문자열의 길이가 1이고, '.'이 첫 문자이면, 무한루프를 종료한다.

문자열의 인덱스 i:  0 ~ 문자열길이를 탐색한다.

s[i] 가 ( 또는 [이면 스택에 해당 문자를 push하고,

s[i] 가 ) 또는 ] 일 때 스택이 비어있지 않고, 스택의 최상단이 해당 문자의 괄호쌍이면 스택에서 pop한다. 만약 그렇지 않다면 no를 출력하도록 break한다.