[백준 17413번] 단어 뒤집기 2

17413번: 단어 뒤집기 2 (acmicpc.net)

17413번: 단어 뒤집기 2

 

#include <iostream>
#include <string>
#include <stack>
using namespace std;
void print(stack<char> &s) {
    while (!s.empty()) {
        cout << s.top();
        s.pop();
    }
}
int main() {
    string str;
    getline(cin, str);
    bool tag = false;
    stack<char> s;
    for (char ch : str) {
        if (ch == '<') {
            print(s);
            tag = true;
            cout << ch;
        } else if (ch == '>') {
            tag = false;
            cout << ch;
        } else if (tag) {
            cout << ch;
        } else {
            if (ch == ' ') {
                print(s);
                cout << ch;
            } else {
                s.push(ch);
            }
        }
    }
    print(s);
    cout << '\n';
    return 0;
}