구현/단어 뒤집기2

Q·2021년 8월 31일
0

알고리즘/백준

목록 보기
33/70

문제 설명


문제

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

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

알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져 있다.
문자열의 시작과 끝은 공백이 아니다.
'<'와 '>'가 문자열에 있는 경우 번갈아가면서 등장하며, '<'이 먼저 등장한다. 또, 두 문자의 개수는 같다.
태그는 '<'로 시작해서 '>'로 끝나는 길이가 3 이상인 부분 문자열이고, '<'와 '>' 사이에는 알파벳 소문자와 공백만 있다. 단어는 알파벳 소문자와 숫자로 이루어진 부분 문자열이고, 연속하는 두 단어는 공백 하나로 구분한다. 태그는 단어가 아니며, 태그와 단어 사이에는 공백이 없다.

입력

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

출력

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


문제링크

전체 코드


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

import sys
from collections import deque

q = deque()
stack = []
result = ''
state = True # 상태체크 위한 변수

input = sys.stdin.readline

s = list(input().strip())

for i in range(len(s)):
    if s[i] == "<": # '<' 입력 받으면
        while stack: # 그 전에 stack에 저장되어 있던 것들 결과값에 저장
            result += stack.pop()
        q.append(s[i]) # '<' 큐에 저장
        state = False # <, > 안의 문자들은 큐에 저장해야 하므로 상태변수 False로 변경

    elif s[i] == ">": # '>' 입력 받으면
        q.append(s[i]) # 큐에 저장
        state = True # <, > 괄호가 끝났으므로 상태변수 True
        while q:     # 괄호 안에 있던 문자 모두 결과 값에 저장
            result += q.popleft() # 순서 그대로 나와야 하므로 deque에서 popleft()

    elif s[i] == " ": # 공백 입력 받았는데
        if state:   # 괄호 밖의 상태이면
            while stack:    # stack에 저장되어 있는것들 다 결과값에 저장
                result += stack.pop() # 공백 전의 문자들은 결과값에 저장해 준 뒤 공백 저장
            result += " "
        else:       # <>괄호 안의 상태라면
            q.append(s[i])  # 큐에 공백 저장해준 뒤
            while q:        # 큐에 저장되어 있는 문자들을 결과값에 저장
                result += q.popleft()
        
    else:           # 입력받은 것이 문자이면
        if state:   # 괄호 밖의 상황
            stack.append(s[i])  # 스택에 저장
        else:       # 괄호 안의 상황
            q.append(s[i])      # 큐에 저장
        
while stack: # 혹시 스택에 남아있는 문자가 있다면
    result += stack.pop()   # 모두 결과 값에 저장

print(result)

해결 방법

주석을 잘 달아 두었다. 코드를 보면서 이해하길!!

profile
Data Engineer

0개의 댓글