[Algorithm] AC

[hwangJi-dev]

💬 문제

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

💬 Idea

• 함수 수행시 R이 나올 때마다 reverse를 수행하고, D가 나올 때마다 removeFirst()를 수행해주었더니 시간 초과가 발생했다.
• 따라서 투포인터 방식을 활용해서 풀이해주었다.
  • R을 수행할 차례라면 isReverse flag를 변경해준다
  • D를 수행할 차례라면
    1. pointer가 reversepointer보다 크다면 → error인 경우이므로 error를 출력하고 continue문을 수행시킨다. (labeled로 인해 중첩 for문 같이 영향)
    2. isReversed flag 상태라면 → reversepointer를 좁혀준다.
    3. isReversed flag 상태가 아니라면 → pointer 범위를 늘려준다.
  • RD 함수 종료 후
    • pointer가 reversepointer보다 크다면 빈 배열을 의미하므로 []를 출력한다.
    • 그렇지 않다면 pointer부터 reversepointer까지 배열을 slice한 후 출력 조건에 맞게 배열을 맵핑한 뒤 출력한다.
      • (배열 자체를 출력할 경우 [1,2,3] 과 같이 출력되지 않고 [1, 2, 3] 과 같이 출력되므로 답이 틀리다고 나오므로 주의하자.)

💬 풀이

import Foundation

func solution5430() {
 let t = Int(readLine()!)!
 
outloop: for _ in 0..<t {
 let function = readLine()!
 let n = Int(readLine()!)!
 let arr = readLine()!.trimmingCharacters(in: ["[", "]"]).split(separator: ",").map({ Int(String($0))! })
 var pointer = 0
 var reversepointer = n - 1
 var isReverse = false
 
 for i in function {
 if i == "R" {
 isReverse = !isReverse
 } else {
 if pointer > reversepointer {
 print("error")
 continue outloop
 }
 
 if isReverse {
 reversepointer -= 1
 } else {
 pointer += 1
 }
 }
 }
 
 if pointer > reversepointer {
 print("[]")
 } else {
 var ans = arr[pointer...reversepointer]
 if isReverse {
 ans.reverse()
 }
 print("[" + ans.map({ String($0) }).joined(separator: ",") + "]")
 }
 }
}

💬 알게된 문법

• for문에 레이블을 붙여 사용하면 중첩 for문도 한꺼번에 탈출이 가능하다.