관리 메뉴

진취적 삶

단어 뒤집기 본문

알고리즘/자료구조 1 연습

단어 뒤집기

hp0724 2023. 6. 29. 12:29

bookmark

문제

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

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

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

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

입력

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

출력

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

나의풀이

s = input()
left_sign = [i for i in range(len(s)) if s[i] == "<"]
right_sign = [i for i in range(len(s)) if s[i] == ">"]

for pair in zip(left_sign, right_sign):
    start_index, last_index = pair
    substring = s[start_index : last_index + 1]

꺽쇠의 인덱스값을 찾아서 zip 형태로 뽑아내서 할라했으나 실패

모범답안

s = input()
reversed_string = ""
stack = []
is_tag = False

for char in s:
    if char == "<":
        is_tag = True
        while stack:
            reversed_string += stack.pop()
        # < 추가
        reversed_string += char
    elif char == ">":
        is_tag = False
        reversed_string += char
        # tag가 열려있으면 그냥 넣기
    elif is_tag:
        reversed_string += char
        # tag가 닫혀있을때
    else:
        # 띄어쓰기 만나면 그전까지 문자열 뒤집기
        if char == " ":
            while stack:
                reversed_string += stack.pop()
            reversed_string += char
        else:
            stack.append(char)
while stack:
    reversed_string += stack.pop()

print(reversed_string)
  1. “<”을 찾으면 그 전까지 stack에 쌓인거 pop으로 뒤집기 그리고 is_tag True
  2. “>” 을 찾으면 is_tag false로 만들고 “>” 추가
  3. is_tag가 열린 상태면 문자열 그대로 추가
  4. is_tag가 닫힌 상태인경우 태그 밖에 이니깐 문자를 뒤집어야 한다. 띄어쓰기 만나면 그동안 stack 에 쌓인거 pop 하기 띄어쓰기 아닌경우 문자 뒤집기를 위한 stack 에 넣기
  5. 태그가 없는 경우 그냥 stack pop 통해서 문자열 뒤집기

'알고리즘 > 자료구조 1 연습' 카테고리의 다른 글

오등큰수  (0) 2023.06.29
오큰수  (0) 2023.06.29
쇠막대기  (0) 2023.06.29