4949: 균형잡힌 세상

-Class2 : Silver5

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

4949번: 균형잡힌 세상

 

-결과

 

-코드

#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <stack>
#define _CRT_SECURE_NO_WARNINGS
using namespace std;

int main(void) {

    while (1) {
        stack<char> s;
        int i;
        char str[101];
        int check = 1;

        cin.getline(str, 101);
        int len = strlen(str);

        if (str[0] == '.') break;

        for (i = 0; i < len; i++) {
            if (str[i] == '(' || str[i] == '[') {
                s.push(str[i]);
            }
            else if (str[i] == ')') {
                if (!s.empty() && s.top() == '(') {
                    s.pop();
                }
                else {
                    check = 0;
                    break;
                }
            }
            else if (str[i] == ']') {
                if (!s.empty() && s.top() == '[') {
                    s.pop();
                }
                else {
                    check = 0;
                    break;
                }
            }
        }

        if (check && s.empty()) { printf("yes\n"); }
        else { printf("no\n"); }
    }
}

 

-풀이

순서대로 입력받은 문자열이 균형을 이루고 있으면 yes를 아니면 no를 출력하는 문제이다.

이 문제는 스택으로 구현하는 문제이다. (가 입력되면 스택에 push를 한다. )가 입력 됐는데 스택이 비어있으면 check값을 0으로 변경시킨 후 반복문을 멈춘다. 만약 스택이 비어있지 않으면 스택의 top 값이 (이면 pop을 한다. ('[ ]'의 경우도 같게 함) 결국 스택이 비어있고 check가 1이면 yes, 나머지 경우는 no를 출력하게 한다.