[BOJ] (Swift) 17413 - 단어 뒤집기 2

17413번: 단어 뒤집기 2

---

 

문제 요약

• 문자열 s에서 tag를 제외한 단어를 뒤집은 문자열을 출력하라.

 

풀이 1 - flag 변수 사용

• 변수 flag로 문자 tag를 체크해 문자 char이 tag일 경우 result에 넣어주고 tag가 아닐 경우에는 stack에 넣어준다.
• "<" 이 나오면 그전까지 stack에 있던 문자를 pop 해서 result에 추가해준다.
• 두 단어 사이의 공백이 나왔을 경우에는 stack에 있던 문자를 pop 해서 result에 추가해주고, 공백 또한 result에 추가한다.
• for문이 종료된 후 stack에 남아있던 문자를 pop해 result에 넣어주는 걸 잊지 말자

import Foundation

let input = readLine()!

var stack: [Character] = []
var result = ""
var flag: Bool = false

for char in input {
    if char == "<" {
        flag = true
        while !(stack.isEmpty) {
            result.append(stack.popLast()!)
        }
        result.append(char)
    } else if char == ">" {
        flag = false
        result.append(char)
    } else if flag {
        result.append(char)
    } else {
        if char == " " {
            while !(stack.isEmpty) {
                result.append(stack.popLast()!)
            }
            result.append(char)
        } else {
            stack.append(char)
        }
    }
}

if !stack.isEmpty {
    result.append(String(stack.reversed()))
}

print(result)