[백준/Silver III] 단어 뒤집기 2 - 17413

분류

자료 구조, 구현, 스택, 문자열

 

문제 설명

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

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

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

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

입력

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

출력

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

import Foundation

let input = readLine()!
var result = ""
var flipWord = ""
var inTag = false

for char in input {
    if char == "<" {
        result += flipWord.reversed()
        flipWord = ""
        result.append(char)
        inTag = true
    } else if char == ">" {
        inTag = false
        result += flipWord.reversed()
        flipWord = ""
        result.append(char)
    } else if char == " " {
        result += flipWord.reversed()
        flipWord = ""
        result.append(char)
    } else if inTag {
        result.append(char)
    } else {
        flipWord.append(char)
    }
}

result += flipWord.reversed()

print(result)

코드 설명:

1. S 문자열을 한 문자씩 순회합니다.
2. < 태그의 시작을 만나면, 그동안 모아둔 단어를 뒤집어서 결과에 추가하고, isTag 플래그를 설정합니다.
3. > 태그의 끝을 만나면, isTag 플래그를 해제합니다.
4. 태그 내부의 문자(isTag가 true인 경우)는 그대로 결과에 추가합니다.
5. 단어(isTag가 false인 경우)는 공백을 만나면 뒤집어서 결과에 추가하고, 단어를 초기화합니다.
6. 순회가 끝난 후, 남아있는 단어가 있으면 뒤집어서 결과에 추가합니다.

시간복잡도 분석:

• 문자열을 한 번 순회하므로 시간복잡도는 O(n)입니다.
• 각 문자에 대해 상수 시간 내의 작업을 수행하므로 전체 시간복잡도는 여전히 O(n)입니다.