import java.util.Stack

fun main() {
    fun handle(input: List<String>, handler: (Array<Stack<Char>>, Int, Int, Int) -> Unit): String {
        val borderLine = input.indexOf("")

        val stacks = Array<Stack<Char>>(input[borderLine - 1].count { it.isDigit() } + 1) { Stack() }
        for ((colIndex, value) in input[borderLine - 1].withIndex().filter { it.value.isDigit() }) {
            val id = value.digitToInt()
            for (rowIndex in borderLine - 1 downTo 0) {
                input[rowIndex][colIndex]
                    .takeIf { it.isUpperCase() }
                    ?.run(stacks[id]::push)
            }
        }

        val re = Regex("move (\\d+) from (\\d+) to (\\d+)")
        for (i in borderLine + 1 until input.size) {
            val result = re.find(input[i])!!
            val (num, from, to) = result.groupValues
                .withIndex().filter { it.index > 0 }.map { it.value.toInt() }
            handler(stacks, num, from, to)
        }
        return stacks.filter { it.isNotEmpty() }.map { it.peek() }.joinToString("")
    }

    fun part1(input: List<String>): String {
        return handle(input) { stacks, num, from, to ->
            repeat(num) {
                stacks[to].push(stacks[from].pop())
            }
        }
    }

    fun part2(input: List<String>): String {
        val tempStack = Stack<Char>()
        return handle(input) { stacks, num, from, to ->
            repeat(num) {
                tempStack.push(stacks[from].pop())
            }
            repeat(num) {
                stacks[to].push(tempStack.pop())
            }
        }
    }

    val testLines = readLines(true)
    check(part1(testLines) == "CMZ")
    check(part2(testLines) == "MCD")

    val lines = readLines()
    println(part1(lines))
    println(part2(lines))
}
