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

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

 

4949번: 균형잡힌 세상

#문제 간단 정리

흔한 괄호문제의 응용

스택을 활용한다.

 

#문제 해결 방법

스택을 활용해 괄호일 경우 스택에 넣어주고 

같은 괄호일경우에 pop 

비어있는 경우나 괄호가 닫는괄호가 아닐경우 오답으로 간주,

 

수많은 스택 활용 괄호 문제중에 하나이다.

 

getline 으로 한줄 전체를 읽어오는 것을 활용하자

그리고 가독성좀 생각해야겠다..

 

#전체 코드

#include <iostream>
#include <vector>
#include <algorithm>
#include <climits>
#include <map>
#include <string>
using namespace std;



int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    vector<string> lines;
    vector<bool> answers;

    while (true) {
        string input;
        getline(cin, input);

        if (input == ".") {
            break;
        }

        vector<char> brackets;
        bool flag = true;

        for (int i = 0; i < input.size(); i++) {

            if (input[i] == '(') {
                brackets.push_back('(');
            }
            else if (input[i] == ')') {
                if (brackets.empty()) {
                    flag = false;
                    break;
                }
                else if (brackets.back() == '(') {
                    brackets.pop_back();
                }
                else if (brackets.back() != '(') {
                    flag = false;
                    break;
                }
            }
            else if (input[i] == '[') {
                brackets.push_back('[');
            }
            else if (input[i] == ']') {
                if (brackets.empty()) {
                    flag = false;
                    break;
                }
                else if (brackets.back() == '[') {
                    brackets.pop_back();
                }
                else if (brackets.back() != '[') {
                    flag = false;
                    break;
                }
            }

        }
        
        if (!brackets.empty()) {
            flag = false;
        }

        if (flag) {
            answers.push_back(true);
        }
        else {
            answers.push_back(false);
        }
    }

    for (bool a : answers) {
        if (a == true) {
            cout << "yes" << '\n';
        }
        else {
            cout << "no" << '\n';
        }
    }


    return 0;
}

리팩토링한 코드

#include <iostream>
#include <vector>
#include <map>
using namespace std;

// 괄호의 짝을 확인하는 함수
bool isValidBracketSequence(const string& input) {
    vector<char> stack;
    map<char, char> bracketPairs = {{')', '('}, {']', '['}};
    
    for (char ch : input) {
        if (ch == '(' || ch == '[') {
            stack.push_back(ch);
        } else if (ch == ')' || ch == ']') {
            if (stack.empty() || stack.back() != bracketPairs[ch]) {
                return false;
            }
            stack.pop_back();
        }
    }
    return stack.empty();
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    vector<bool> answers;
    string input;

    while (true) {
        getline(cin, input);
        if (input == ".") break;

        answers.push_back(isValidBracketSequence(input));
    }

    for (bool isValid : answers) {
        cout << (isValid ? "yes" : "no") << '\n';
    }

    return 0;
}