[ BOJ ] 17413 : 단어 뒤집기 2 ( SILVER 3 ) / C, Python

문제

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

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

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

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

입력

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

출력

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

풀이 과정

문자열을 순차적으로 탐색하면서..

 

1. < 가 등장했다면 -> 태그가 시작된다는 표시이니, 스택에 쌓여 있는 단어 문자들을 전부 뒤집어 출력한 뒤 태그가 끝날 때 까지 순차적으로 문자들을 출력한다. ( 태그는 단어가 아니므로 )

2. > 가 등장했다면 -> 태그가 끝났다는 표시이다.

3. 태그가 아닌데 공백이 등장했다면 -> 지금까지 스택에 쌓여 있는 단어 문자들을 전부 뒤집어 출력한다.

 

태그는 단어가 아니므로, 태그 안이라면 문자를 그냥 출력하고,

태그가 아니라 단어라면 문자를 스택에 저장해놓은 뒤 이후 공백이 등장했거나 태그가 등장했거나 문자열의 끝에 도달했을 때에 한꺼번에 꺼내어 뒤집어 출력한다.

C

#include <stdio.h>
#include <string.h>

char str[100005];
char stack[100005];
int stackSize = 0;
int main(void) {
    gets(str);

    int tag_mode = 0;
    for (int i = 0; i < strlen(str); i++) {
        if (str[i] == '<') {
            while (stackSize)
                printf("%c", stack[--stackSize]);
            tag_mode = 1;
            printf("%c", str[i]);
        }
        else if (str[i] == '>') {
            tag_mode = 0;
            printf("%c", str[i]);
        }
        else if (tag_mode == 0 && str[i] == ' ') {
            while (stackSize)
                printf("%c", stack[--stackSize]);
            printf(" ");
        }
        else if (tag_mode == 1) printf("%c", str[i]);
        else stack[stackSize++] = str[i];
    }

    while (stackSize)
        printf("%c", stack[--stackSize]);

    return 0;
}

Python

S = input()
S_reversed = ''
while S:
    left_bracket = S.find('<')
    right_bracket = S.find('>')
    if left_bracket == -1:
        break
    if left_bracket != 0:
        temp = S[:left_bracket].split()
        if S[:left_bracket][0] == ' ':
            S_reversed += ' '
        for i in range(len(temp)):
            S_reversed += ''.join(reversed(temp[i]))
            if i != len(temp) - 1:
                S_reversed += ' '
        if S[:left_bracket][-1] == ' ':
            S_reversed += ' '
    S_reversed += S[left_bracket:right_bracket + 1]
    S = S[right_bracket + 1:]


if S != '':
    for element2 in S.split():

        S_reversed += ''.join(reversed(element2))
        S_reversed += ' '
print(S_reversed)