프로그래머스 - 택배상자 Swift (스택, 구현)

https://school.programmers.co.kr/learn/courses/30/lessons/131704

프로그래머스

---

풀이 및 소스코드

보조 컨베이어 벨트는 뒤에서만 뺄 수 있으므로 대놓고 스택문제다.

배열로 스택을 구현하고 아래의 설명처럼 구현하면 된다.

import Foundation

func solution(_ order:[Int]) -> Int {
    var target = 0
    var stack = [Int]()
    var truck = [Int]()
    
    //상자는 1부터 order의 갯수만큼 들어온다.
    for num in 1...order.count{
        
        //만약 상자가 타겟과 다르다면
        if order[target] != num{
            //스택이 비어있을 경우 스택에 넣고 다음상자
            if stack.isEmpty{
                stack.append(num)
                continue
            }
            //스택이 비어있지 않고 스택의 마지막이 타겟과 같다면
            while !stack.isEmpty && stack.last! == order[target]{
                //트럭에 상자를 추가하고 스택을 감소시킨후 타겟을 증가시킨다.
                truck.append(stack.removeLast())
                target += 1
            }
            
            //스택의 마지막이 타겟과 다르다면 스택에 추가하고 다음상자.
            if stack.last != order[target]{
                stack.append(num)
                continue
            }
        }
        
        //타겟과 넘버가 같으면 트럭에 추가하고 타겟증가
        if num == order[target]{
            truck.append(num)
            target += 1
        }
        
    }
    
    //원래 컨테이너의 상자를 다 옮기고, 보조트레일러의 상자를 정리
    while !stack.isEmpty && stack.last! == order[target]{
        truck.append(stack.removeLast())
        target += 1
    }
    
    return truck.count
}