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

단어 뒤집기 2

 

문제

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

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

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

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

입력

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

출력

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

---

문제풀이

처음에 인풋 입력을 어떻게 받아야 할지 당황했다. 맨날 cin만 쓰다보니 몰랐다.

getline을 이용해서 한 라인을 한번에 받아올 수 있다.

스택을 쓰지 않아도 풀 수 있을만큼 간단하다. 태그가 겹치지 않기 때문이다.

 

문자열을 하나씩 읽으며 < 가 나올경우나올 경우 ' '(공백)이 나올 경우, 그냥 문자가 나올 경우로 구분한다.

문자가 나올경우 나중에 출력하게 string 객체에 담아둔다.

<가 나올경우 이전에 저장된 단어를 뒤집어서 출력하고, >가 나올 때까지 출력한다.

' '가 나올경우 이전에 저장된 단어가 존재한다면 뒤집어서 출력하고, ' '를 출력한다.

반복문이 끝나면 저장된 단어가 있을 수 있으니 그것을 확인해서 있으면 출력한다.

 

#include <iostream>
#include <string>

using namespace std;

// 저장된 단어 몽땅 출력
void flush(string& s) {
    if(!s.empty()) {
        for(int j=s.size()-1; j>=0; j--)
            cout << s[j];
    }
}

int main() {
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    string s;
    getline(cin, s);
    string c = "";
    for(int i=0; i<s.size(); i++) {
        // < 일경우 이전에 저장된 단어 출력, > 까지 출력
        if(s[i] == '<') {
            flush(c);
            while(s[i] != '>') {
                cout << s[i++];
            }
            cout << s[i];
            c = "";
        }
        // ' '일 경우 이전에 저장된 단어 있으면 출력, ' '출력
        else if(s[i] == ' '){
            flush(c);
            cout << s[i];
            c = "";
        }
        else {
            c += s[i];
        }
    }
    flush(c);

    return 0;
}