[17413] 단어 뒤집기2

🌿 문제

[17413] 단어 뒤집기2

 

🌿 해결 방법

경우는 크게 3가지로 나누어서 보면 된다.

1. '<' 기호를 만난 경우
2. 단어 사이에 있는 공백 문자를 만난 경우
3. 소문자 or 숫자를 만난 경우

첫번째 경우에는 스택에 저장된 문자들이 있는지 확인하고, 있다면 하나씩 뽑아서 출력한다.

스택은 선입후출 방식이므로 하나씩 뽑아서 출력하면 그게 거꾸로 출력하는 것이 된다.

그리고 닫는 태그 '>'를 만날 때까지 나오는 모든 소문자, 공백, 숫자를 그대로 출력한다.

 

두번째 경우는 단어 하나가 끝났다는 뜻이기 때문에 스택을 점검하고 비어있지 않다면 모두 출력해준다.

 

세번째 경우는 그냥 스택에 바로 추가한다.

 

마지막으로 스택이 비어있는지 다시 한번 확인한 후, 남아있는 문자가 있다면 출력해준다.(마지막 단어)

한 가지 주의할 부분은 입력 문자 s에 공백이 있을 수 있기 때문에 getline을 통해 반드시 공백을 포함해서 입력받아야 한다.

 

🌿 내 코드

// [17413] 단어 뒤집기2
// 구현, 문자열, 스택
#include <iostream>
#include <stack>
using namespace std;

stack<char> wordStack;

void printStack()
{
    while(!wordStack.empty()) {
        cout << wordStack.top();
        wordStack.pop();
    }
}

void go(string s)
{
    for(int i = 0; i < s.size(); i++) {
        if(s[i] == '<') {
            printStack();
            cout << s[i];
            while(s[i] != '>') {
                cout << s[++i];
            }
        }
        else if(s[i] == ' ') {  
            printStack();
            cout << s[i];
        }   
        else { // 소문자, 숫자
            wordStack.push(s[i]);
        }   
    }
    printStack();
}

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

    char s[100005];
    cin.getline(s, 100005);
    go(s);
    return 0;
}