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

문제 개요

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

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

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

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

입력

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

출력

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

아이디어

• 처음부터 끝까지 보면서 공백이 들어올 때까지 스택에 넣는다.
• 공백이 나오면 팝하면서 출력함.
• 만약 <가 나오면 >가 나올때까지 그대로 출력한다.

그대로 구현했는데

1. 공백 포함 한 줄 입력받기는 cin>>str이 아니라 getline을 써야 함 (이것때문에 오류남)
2. 공백 그대로 출력

이 부분만 좀 헤맸다..

코드

#include<iostream>
#include<algorithm>
#include<vector>
#include<string>
#include<math.h>
#include<stack>
using namespace std;

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

	stack<char> s;
	string str;
	getline(cin, str); //공백 포함 문자열 입력받기

	/* 처음부터 끝까지 보면서 공백이나 <가 나올 때까지 스택에 넣는다.
	공백이나 <가 나오면 팝하면서 출력함.
	만약 '<'가 나오면 '>'가 나올때까지 그대로 출력한다.*/
	int i;
	for (i = 0; i < str.length(); i++) {
		if (!s.empty() &&
			(str[i] == '<' || str[i] == ' ' )) { // 스택 비우는 조건
			
			while (!s.empty()) {
				cout << s.top(); s.pop();
			}
		}

		if (str[i] == ' ') {
			cout << ' ';
			continue;
		}

		if (str[i] == '<') { // >가 나올때까지 그대로 출력함
			while (1) {
				cout << str[i];
				i++;
				if (str[i] == '>') break;
			}
			cout << ">";
		}

		else s.push(str[i]);
		// <태그>안의 내용은 스택에 넣지 않는다
		
	}

	if (!s.empty()) { //마지막까지 갔을때 마지막 단어가 출력되지 않았을떄
		while (!s.empty()) {
			cout << s.top();
				s.pop();
		}
	}


	return 0;
}