백준[17413] 단어 뒤집기 2

문제 설명

문제 링크

http://icpc.me/17413

17413번: 단어 뒤집기 2

풀이

스택을 이용하여 풀이할 수 있다.

문자열을 처음부터 읽으면서 아무것도 만나지 않았을 때는 stack에 담다가 <을 만나거나 공백을 만나면 stack의 모든 문자열을 pop 하면 된다. 그리고 만약 <을 만났다면 >를 만날 때까지 그냥 출력한다.

코드

#include <cstdio>
#include <stack>
#include <cstring>

using namespace std;

char s[100005];
stack<char> st;

int main() {
    scanf("%[^\n]", s);
    int sz = strlen(s);
    bool flag = false;
    for (int i = 0; i < sz; i++) {
        if (s[i] == '<') {
            while (!st.empty()) {
                printf("%c", st.top());
                st.pop();
            }
            printf("<");
            flag = true;
        }
        else if (s[i] == '>') {
            printf(">");
            flag = false;
        }
        else if (flag) printf("%c", s[i]);
        
        else if (s[i] == ' ') {
            while (!st.empty()) {
                printf("%c", st.top());
                st.pop();
            }
            printf(" ");
        }
        else st.push(s[i]);
    }
    while (!st.empty()) {
        printf("%c", st.top());
        st.pop();
    }
}