백준 17413 단어 뒤집기 2 c++ (문자열)

문제 출처 : https://www.acmicpc.net/problem/17413

17413번: 단어 뒤집기 2

 

문제

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

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

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

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

입력

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

출력

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

알고리즘 분류

• 구현
• 문자열

풀이

어렵진 않지만 틀리기 쉬운 문자열 문제이다.

태그 안에 있는 단어들은 건들지 말고, 바깥에 공백 및 여는 태그'<' , 닫는 태그'>'로 분류되어 있는

단어들만 뒤집어 주면 된다.

우선 input의 0번 인덱스부터 마지막까지 검사하며,

태그가 열려있을 때는, answer에 그대로 문자를 추가하고,

'>'문자를 입력받을 시 answer에 '>'를 추가하고 태그를 닫아준다.

태그가 닫혀있을 때는 문자가 '<'라면 태그를 열어주고, 현재 word(태그 바깥에 있는 단어)가 있다면, word를 초기화한 후, answer에 단어를 뒤집어서 추가한 다음, answer에 '<'를 추가한다.

태그가 닫혀있을 때 문자가 공백이라면 현재 word가 있는지 확인하고 answer에 단어를 뒤집어서 추가한 다음, word를 초기화한 후, answer에 공백을 추가한다.

태그가 닫혀있을 때 문자가 단어의 일부라면 word에 추가하여 단어를 완성해 나간다.

input의 마지막까지 탐색 후, word가 남아있다면 answer에 단어를 뒤집어서 추가한다.

코드

#include <iostream>
#include <string>
#include <sstream>
#include <vector>
using namespace std;

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);


	string input;
	getline(cin, input);
	char tag = '>';
	string word = "";
	string answer = "";
	for (int i = 0; i < input.length(); i++) {
		char ch = input[i];
		//태그가 열려 있을때
		if (tag == '<') {
			if (ch == '>') {
				tag = '>';
			}
			answer += ch;
		}
		//태그가 닫혀 있을때
		else {
			if (ch == '<') {
				tag = '<';
				if (word.length() != 0) {
					for (int j = word.length() - 1; j >= 0; j--) {
						answer += word[j];
					}
					word = "";
				}
				answer += ch;
			}
			else if (ch == ' ') {
				if (word.length() != 0) {
					for (int j = word.length() - 1; j >= 0; j--) {
						answer += word[j];
					}
					word = "";
					answer += ' ';
				}
			}
			else {
				if (ch != '>') {
					word += ch;
				}
			}
			
		}
	}
	if (word.length() != 0) {
		for (int j = word.length() - 1; j >= 0; j--) {
			answer += word[j];
		}
	}
	cout << answer;
	return 0;
}