[BaekJoon 17413번] 단어 뒤집기 2(C++)

※주의※
저의 풀이가 정답은 아닙니다.
다른 코드가 더 효율적이거나 좋을 수 있습니다.
언제나 다른 사람의 코드는 참고만 하시기 바랍니다.

 
 

[문제 풀이 사이트]

 
 

17413번: 단어 뒤집기 2

 
 

[문제 설명]

 
 
 

문자열 S가 주어졌을 때, 이 문자열에서 단어만 뒤집으려고 한다.
먼저, 문자열 S는 아래와과 같은 규칙을 지킨다.
 

1. 알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져 있다.
2. 문자열의 시작과 끝은 공백이 아니다.
3. '<'와 '>'가 문자열에 있는 경우 번갈아가면서 등장하며, '<'이 먼저 등장한다. 또, 두 문자의 개수는 같다.

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

 

[입력]

 
 

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

 
 

[출력]

 
 

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

 
 

[입출력 예]

 
 

입력	출력
baekjoon online judge	noojkeab enilno egduj
<open>tag<close>	<open>gat<close>
<ab cd>ef gh<ij kl>	<ab cd>fe hg<ij kl>
one1 two2 three3 4fourr 5five 6six	1eno 2owt 3eerht rruof4 evif5 xis6
<int><max>2147483647<long long><max>9223372036854775807	<int><max>7463847412<long long><max>7085774586302733229
<problem>17413<is hardest>problem ever<end>	<problem>31471<is hardest>melborp reve<end>
<   space   >space space space<    spa   c e>	<   space   >ecaps ecaps ecaps<    spa   c e>

 
 

[문제 풀이]

 
 

stack과 queue를 활용했던 문제이다.

태그('<', '>')가 포함된 값은 단어에 해당되지 않기 때문에 뒤집을 필요가 없다.
그래서 FIFO(First In First Out)인 queue를 사용하여 값들을 저장하도록 했다.

태그가 아닌 값(알파벳 또는 숫자, 공백으로 이루어짐)은 단어에 해당되기 때문에 뒤집어야 한다.
그래서 FILO(First In Last Out)인 stack을 사용하여 값들을 저장하도록 했다.

내가 작성한 알고리즘은 다음과 같다.

1. 태그의 시작인 '<'는 태그를 시작할 때 또는 단어의 끝에 올 수 있다.
그래서 '<'가 나타났을 때 단어가 있을지도 모르는 stack을 확인하여 정답 문자열에 넣어준다.
(stack 이기 때문에 자동으로 뒤집어서 표현이 된다.)

2. 태그의 마지막인 '>'는 태그를 끝낼 때 올 수 있다.
그래서 '>'가 나타났을 때 태그에 포함되는 값들이 들어있는 queue를 확인하여 정답 문자열에 넣어준다.
(queue 이기 때문에 뒤집어서 표현이 되지 않는다.)

3. ' ' 공백이 왔을 때는 이전 단어가 있을지도 모르기 때문에 stack을 확인하여 정답 문자열에 넣어준다.
그리고 ' ' 공백을 정답 문자열에 넣어준다.

4. 그 외 알파벳이나 숫자가 왔을 때 태그의 값들이 저장되어있는 queue가 비어있지 않다면, queue에 값이 저장이 되고 비어있다면 stack에 저장이 된다.

5. 그리고 반복문이 끝나고 이전 단어가 있을지도 모르기 때문에 stack을 확인하여 stack에 저장된 값들을 정답 문자열에 넣어준다.
 
 

[최종 코드]

 
 

[GitHub]

 
 

GitHub - SmallPeanutPark/BAEKJOON: Study BaekJoon for Coding Test

 
 

#include <bits/stdc++.h>
using namespace std;

int main(void) {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    string answer;
    string str;
    getline(cin, str);
    queue<char> tag;
    stack<char> reverse_word;
    for(char element : str) {
        if((element == '<') || (element == '>')) {
            tag.push(element);
            if(element == '<') {
                while(!reverse_word.empty()) {
                    answer += reverse_word.top();
                    reverse_word.pop(); 
                }
            } else if(element == '>') {
                while(!tag.empty()) {
                    answer += tag.front();
                    tag.pop();
                }
            }
        } else {
            if(!tag.empty()) {
                tag.push(element);
            } else if(element == ' ') {
                while(!reverse_word.empty()) {
                    answer += reverse_word.top();
                    reverse_word.pop();
                }
                answer += element;
            } else {
                reverse_word.push(element);
            }
        }
    }

    while(!reverse_word.empty()) {
        answer += reverse_word.top();
        reverse_word.pop();
    }

    cout << answer;
    return 0;
}