(Python/파이썬) - 백준(BOJ) 17413번 : 단어 뒤집기 2

https://www.acmicpc.net/problem/17413

17413번: 단어 뒤집기 2

---

• 문제 : 

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

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

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

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

 

 

 

---

• 풀이 :

"<" 와 ">" 태그가 나왔을 때 revFlag(뒤집기 여부를 확인하는 플래그)를 변경해주었습니다.

또한, stack을 이용하여 revFlag가 True일 경우 stack이 빌 떄 까지 pop() 해주어 이 값을 정답 스트링에 더해주었습니다.

처음에 공백을 기준으로 뒤집는 조건을 빼먹어 한 번 틀렸습니다.

예를 들어,

a b c d e (input)

a b c d e (output)

이 올바른 정답인데 

저의 답은

e d c b a로 나와 틀렸습니다.

 

공백을 체크해주는 조건을 추가해주어 통과할 수 있었습니다.

 

 

 

 

---

• 소스코드 : 

string = []
revFlag = True
answer = ""
for i in input():
    if i == "<":
        while string:
            answer += string.pop()
        answer += i
        revFlag = False
    elif i == ">":
        answer += i
        revFlag = True
    elif i == " " and revFlag == True:
        while string:
            answer += string.pop()
        answer += i
    else:
        if revFlag:
            string.append(i)
        else:
            answer += i
while string:
    answer += string.pop()
print(answer)