[C++][BOJ/백준] 17413 단어 뒤집기 2

 

 

Baekjoon Online Judge

 

클릭 시 문제로 이동

 

 

[문제]

문자열 S가 주어졌을 때, 이 문자열에서 단어만 뒤집으려고 한다.

먼저, 문자열 S는 아래와과 같은 규칙을 지킨다.

 

1. 알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져 있다.

2. 문자열의 시작과 끝은 공백이 아니다.

3. '<'와 '>'가 문자열에 있는 경우 번갈아가면서 등장하며, '<'이 먼저 등장한다. 또, 두 문자의 개수는 같다.

 

태그는 '<'로 시작해서 '>'로 끝나는 길이가 3 이상인 부분 문자열이고, '<'와 '>' 사이에는 알파벳 소문자와 공백만 있다. 단어는 알파벳 소문자와 숫자로 이루어진 부분 문자열이고, 연속하는 두 단어는 공백 하나로 구분한다. 태그는 단어가 아니며, 태그와 단어 사이에는 공백이 없다.

 

 

[입력]

첫째 줄에 문자열 S가 주어진다. S의 길이는 100,000 이하이다.

 

 

[출력]

첫째 줄에 문자열 S의 단어를 뒤집어서 출력한다.

 

 

---

 

[코드]

#include <iostream>
#include <string>
#include <stack>
using namespace std;

string res;
stack<char> st;

void stackPop()
{
    while (!st.empty())
    {
        res += st.top();
        st.pop();
    }
}

int main()
{
    string s;
    bool tag = false;

    getline(cin, s);

    for (int i = 0; i < s.length(); i++)
    {
        if (s[i] == '<')
        {
            stackPop();
            tag = true;
        }
        else if (s[i] == '>')
        {
            res += s[i];
            tag = false;

            continue;
        }
        else if (s[i] == ' ')
        {
            stackPop();
            res += ' ';

            continue;
        }

        if (tag)
            res += s[i];
        else
            st.push(s[i]);
    }

    stackPop();

    cout << res << '\n';

    return 0;
}