P41_3.kt

package ch12

import java.util.*

class P41_3 {
    fun findItinerary(tickets: List<List<String>>): List<String> {
        val fromToMap: MutableMap<String, PriorityQueue<String>> = mutableMapOf()

        // 여행 일정을 from → to 형태의 그래프로 구성
        for (ticket in tickets) {
            // 값이 존재하지 않을 경우 빈 우선순위 큐 생성
            fromToMap.putIfAbsent(ticket[0], PriorityQueue())
            // 목적지 to 추가, 우선순위 큐이므로 to는 항상 사전 어휘순으로 정렬된다.
            fromToMap[ticket[0]]!!.add(ticket[1])
        }

        val results: MutableList<String> = LinkedList()
        val stack: Deque<String> = ArrayDeque()
        // 출발지 삽입
        stack.push("JFK")
        while (!stack.isEmpty()) {
            // 스택에서 추출될 값을 출발지로 한 도착지 처리
            while (fromToMap.containsKey(stack.first) && !fromToMap[stack.first]!!.isEmpty()) {
                // 여러 도착지 중 사전 어휘순으로 추출해 스택에 삽입
                stack.push(fromToMap[stack.first]!!.poll())
            }

            // 위 반복문이 모두 실행된 이후에는 스택에서 값이 하나씩 추출되어 정답으로 구성된다.
            results.add(0, stack.pop())
        }
        return results
    }
}