[백준 4949] 균형잡힌 세상 (C++/Python)

C++)

#include <iostream>
#include <string>
#include <stack>

using namespace std;

int main(void) {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);

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

		if (str == ".")
			break;

		stack<char> st;
		for (auto ele : str) {
			if (isalpha(ele) || ele == ' ' || ele == '.') 
				continue;

			if (ele == '(' || ele == '[')
				st.push(ele);
			else if (!st.empty() && st.top() == '(' && ele == ')')
				st.pop();
			else if (!st.empty() && st.top() == '[' && ele == ']')
				st.pop();
			else
				st.push(ele);
		}

		if (st.empty())
			cout << "yes\n";
		else
			cout << "no\n";
	}
	return 0;
}

 

 

Python)

import sys

read = sys.stdin.readline

while True:
    temp = read().rstrip()

    if temp == '.':
        break

    st = []
    for x in temp:
        if x.isalpha() or x == ' ' or x == '.':
            continue

        if x == '(' or x == '[':
            st.append(x)
        elif st and st[-1] == '(' and x == ')':
            st.pop()
        elif st and st[-1] == '[' and x == ']':
            st.pop()
        else:
            st.append(x)

    print("no" if st else "yes")