Skip to Content

Advent of Code 2017 - Day 16, in Kotlin

Kotlin solutions to parts 1 and 2 of Advent of Code 2017, Day 16: 'Permutation Promenade'

Posted on

On Day 16 we’ll revisit an past solution for inspiration and save ourselves 7 days of waiting. If you’d rather jump straight to the code, it’s right here.

I’ve challenged myself to post each day’s solution to Github and blog about it. Like last year, I’m going to be solving these problems in Kotlin. I might go back and revise solutions, but I’ll be sure to annotate that fact in these posts.

Problem Input

We are given a set of input for this problem. I’ve loaded this into a String called input, which we will do some further parsing on later.

Day 16, Part 1

You come upon a very unusual sight; a group of programs here appear to be dancing.

There are sixteen programs in total, named a through p. They start by standing in a line: a stands in position 0, b stands in position 1, and so on until p, which stands in position 15.

The programs’ dance consists of a sequence of dance moves:

  • Spin, written sX, makes X programs move from the end to the front, but maintain their order otherwise. (For example, s3 on abcde produces cdeab).
  • Exchange, written xA/B, makes the programs at positions A and B swap places.
  • Partner, written pA/B, makes the programs named A and B swap places.

For example, with only five programs standing in a line (abcde), they could do the following dance:

s1, a spin of size 1: eabcd. x3/4, swapping the last two programs: eabdc. pe/b, swapping programs e and b: baedc.

After finishing their dance, the programs end up in order baedc.

You watch the dance for a while and record their dance moves (your puzzle input). In what order are the programs standing after their dance?

I’m going to implement our tiny set of instructions as a sealed class to make things easier on myself.

sealed class Dance
class Spin(val amount: Int) : Dance()
class Exchange(val left: Int, val right: Int) : Dance()
class Partner(val left: Char, val right: Char) : Dance()

And we’ll do the somewhat messy parsing of the giant input string into instructions:

class Day16(input: String, private val programNames: String = "abcdefghijklmnop") {

	private val instructions: List<Dance> = parseInput(input)

	private fun parseInput(input: String): List<Dance> =
	    input
	        .split(",")
	        .map { it.trim() }
	        .map {
	            when (it.first()) {
	                's' -> Spin(it.drop(1).toInt())
	                'x' -> {
	                    val (a, b) = it.drop(1).split("/").map { it.toInt() }
	                    Exchange(a, b)
	                }
	                'p' -> {
	                    Partner(it[1], it[3])
	                }
	                else -> throw IllegalArgumentException("Bad input: $it")
	            }
	        }
}

I’ll let you go through the nitty-gritty details on that, but essentially we split our input into individual strings, see what character they start with, and then rip the rest of the string apart and create a subclass of Dance.

And of course, we need to have our instructions evaluated against our programs, so I’ve written a function to evaluate a single instruction. In the case of a Spin instruction, a new CharArray is created and returned. In the case of Exchange and Partner it just returns the programs it was given because those are swapped in place. Also, I’ve written another swap extension function on CharArray, which just swaps two values in place (similar to the version we already have on IntArray).

private fun evaluate(programs: CharArray, instruction: Dance): CharArray =
    when (instruction) {
        is Spin -> {
            (programs.takeLast(instruction.amount) + programs.dropLast(instruction.amount)).toCharArray()
        }
        is Exchange ->
            programs.swap(instruction.left, instruction.right)
        is Partner ->
            programs.swap(programs.indexOf(instruction.left), programs.indexOf(instruction.right))
    }

Solving part one involves running through the instructions a single time and joining the resulting programs list into a string.

fun solvePart1(): String =
    executeInstructions()

private fun executeInstructions(startState: CharArray = initialState): String =
    instructions.fold(startState) { carry, next -> evaluate(carry, next) }.joinToString("")

I use a fold to iterate through the instructions, starting with the start state. Despite the amount of code we wrote to parse the input, I feel this part of the challenge was pretty easy. We’ve earned our star, let’s move on.

Day 16, Part 2

Now that you’re starting to get a feel for the dance moves, you turn your attention to the dance as a whole.

Keeping the positions they ended up in from their previous dance, the programs perform it again and again: including the first dance, a total of one billion (1000000000) times.

In the example above, their second dance would begin with the order baedc, and use the same dance moves:

  • s1, a spin of size 1: cbaed.
  • x3/4, swapping the last two programs: cbade.
  • pe/b, swapping programs e and b: ceadb.

In what order are the programs standing after their billion dances?

Let’s do some quick math. After parsing the instructions, the executeInstructions function takes just under 1 millisecond to run . Let’s round that up to 1ms to account for garbage collection and whatever other factors are at play in a longer run. Do that 1,000,000,000 times, and we have to wait for ALMOST 11 DAYS. Clearly, there has to be a better way to do this.

Let’s think back to the puzzles we’ve already solved. If you look at Day 6 we were asked to identify a cycle in our memory allocation algorithm. Could we use the same thing here? Given 1,000,000,000 runs with only 16 programs and the fact that the person who invented this puzzle clearly didn’t want us to wait a week to solve it, I bet we can.

My approach is going to be:

  1. Create a map called memory to store the state of the programs (hash) after they are run, and the run number we found it in.
  2. Keep executing the full set of instructions and record the hash and run number until we generate a duplicate.
  3. Figure out the size of the cycle and calculate out how far from the end of 1,000,000,000 runs we will be if we eliminate redundant cycles.
  4. Look in the memory for what the 1,000,000,000th record would be, had we run all the way to the end.

When I ran this on my input the cycle was fairly short - 48 runs long starting with the first run. I’m not sure, but it seems logical that the cycle might not always start at 1, so we have to account for that for a general solution.

I’ve decided to model this as a tail recursive function rather than iterate as a loop. Both work fine, both do the same work, I just think the recursive solution looks more elegant.

tailrec fun solvePart2(memory: Map<String, Int> = mapOf(), loopNumber: Int = 0, hash: String = programNames): String {
    return if (hash in memory) {
        // We found it!
        val cycleStart = memory.getValue(hash)
        val offset = (1_000_000_000 % (loopNumber - cycleStart)) - cycleStart
        memory.entries.first { it.value == offset }.key
    } else {
        solvePart2(memory + (hash to loopNumber), loopNumber.inc(), executeInstructions(hash.toCharArray()))
    }
}

Let’s go over it. The first time we call solvePart2, we initialize our memory (the hashes we’ve already seen), a loopNumber (how many times we’ve done this), and a hash (defaulting to the original alphabet). If we don’t find the hash in our memory, we add it to the memory, increment our loop number, execute the instructions against the hash we were given and pass that all, recursively, to ourselves.

Because this is recursive, we need a terminal condition. That is, of course, finding the hash in our memory. When that happens, we calculate our offset by dividing the cycle length into 1,000,000,000, and subtracting the loop number that started the cycle in case the cycle doesn’t start at zero (mine did). Remember: the offset represents how far along the cycle the 1,000,000,000th entry is. Finally, we search through the values in our memory until we find the offset, and there’s our end hash!

This implementation should be tail recursive or the stack will overflow if you have a long enough cycle. For me this runs in about 125ms! We could run that 8,000,000 times in the time it would take to do this without eliminating cycles! I’m sure there are plenty of optimizations that could be made to get this even quicker, if we were so inclined.

We either cheated time or broke the fabric of space (perhaps both?) but it earns us a star in either case, so that’s cool.

I hope you’ve learned something and as always, feedback is welcome!

2^4 days and we’ve earned 2^5 stars with 18 left to go! Only 9 days left!

Further Reading

  1. Index of All Solutions - All solutions for 2017, in Kotlin.
  2. My Github repo - Solutions and tests for each day.
  3. Solution - Full code for day 16.
  4. Advent of Code - Come join in and do these challenges yourself!
  5. Music to Code By - Alice Practice, by Crystal Castles.