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

문제

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

4949번: 균형잡힌 세상

 

풀이

https://hun6728.tistory.com/4 문제의 응용버전이라고 생각해주면 된다. 이 문제에서는 소괄호, 대괄호가 있고 여는 괄호일 때는 PUSH, 닫는 괄호일 때는 POP을 해주면 된다. 또한 주의해야 할 점이 있는데, POP을 할 때 괄호간의 짝이 맞는지도 판별해주어야 한다. 닫는 괄호가 ')' 일때는 스택의 맨 위에 있는 괄호가 '(' 이어야하며, 닫는 괄호가 ']' 일 때는 스택의 맨 위에 있는 괄호가 '[' 이어야 한다. 이를 통해 괄호간의 쌍이 맞는지를 판별해주면 된다.

 

코드

#include<stdio.h>
#include<string.h>
#include<stack>

using namespace std;

#define LEN 300

int main() {
    int i;
    for (;;) {
        char input[LEN];
        stack<char> stack;

        fgets(input, LEN, stdin);
        int len = strlen(input);
        if (len == 2 && input[0] == '.') {
            break;
        }
        for (i = 0; i < len; i++) {
            if (input[i] == '(' || input[i] == '[') {
                stack.push(input[i]);
            }
            if (input[i] == ')' || input[i] == ']') {
                if (stack.empty()) break;
                if (input[i] == ')' && stack.top() == '(') {
                    stack.pop();
                    continue;
                }
                if (input[i] == ']' && stack.top() == '[') {
                    stack.pop();
                    continue;
                }
                break;
            }
        }
        if (i == len && stack.size() == 0) {
            printf("yes\n");
        } else {
            printf("no\n");
        }
    }
    return 0;
}