Advent of Code 2017 - Day 14, in Kotlin
Kotlin solutions to parts 1 and 2 of Advent of Code 2017, Day 14: 'Disk Defragmentation'
On Day 14 we’ll lean on some code we wrote a few days ago, and learn about recursively visiting spots on a grid. 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 14, Part 1
The puzzle text can be found here.
Today’s problem does what problems of years past did, build on previous day’s work. In this case we’re going to reuse our knot hash logic from Day 10
. I’m not going to cover it here, but I refactored it into its own class called KnotHash
already. You can view the code here on GitHub
.
If you did day 10, this shouldn’t look too surprising.
Now that we have a hash, it’s just a matter of creating the 128 rows of our grid using the hash, and converting the hex string to binary. Thankfully BigInteger
can do both of these things for us! Let’s get on with parsing the input:
private val binaryStrings = parseInput(input)
private fun parseInput(input: String): List<String> =
(0..127)
.map { KnotHash.hash("$input-$it") }
.map { BigInteger(it, 16).toString(2).padStart(128, '0') }
Parsing the input is pretty simple, we just loop 128 times and add the digit we generate to our input string as directed. Then we hash it, convert that hex string to a BigInteger
(specifying that it is in base 16/hex), and finally convert it to binary (base 2). Because the string might start with zeros, they’ll get lost in the conversion so we pad the string up to the desired 128 length with 0 (we care about this in part two).
And now to solve, count up the occurrences of 1
:
fun solvePart1(): Int =
binaryStrings.sumBy { it.count { it == '1' }
Thanks to a bunch of code we had already written, this earns us an easy star!
⭐ Day 14, Part 2
The puzzle text can be found here.
I had to read this about four times before I realized what was being asked. We’re numbering regions of contiguous 1
s in the grid. The diagram above shoes their grid number. First we will convert our input from a List<String>
to a List<IntArray>
to make things easier to count later.
private val grid = stringsToGrid()
private fun stringsToGrid(): List<IntArray> =
binaryStrings
.map { s -> s.map { it.asDigit() } }
.map { it.toIntArray() }
So how do we find groups? My thought process is:
- Keep a counter of groups found so far starting at 0.
- Iterate over each row and column.
- If the
x,y
value we’re looking at is a1
, we have at least part of a group.- Mark that spot as 0 so we don’t see it again.
- Increment the counter.
- Given that starting point
x
andy
, recursively find all its neighbors and make them 0.
I’m marking all the groups down to 0 because we’re not asked which group does what, or which is the largest. We’re being asked for a total count which I’m tracking with the counter. We could change them to anything that isn’t 1
, so go nuts.
First, let’s write a function to get all valid neighbors of a given x
and y
:
private fun neighborsOf(x: Int, y: Int): List<Pair<Int, Int>> =
listOf(Pair(x - 1, y), Pair(x + 1, y), Pair(x, y - 1), Pair(x, y + 1))
.filter { it.first in 0..127 }
.filter { it.second in 0..127 }
Now that we have a way to generate valid neighbors that are always in range, lets write our recursive group eliminating function:
private fun markNeighbors(x: Int, y: Int) {
if (grid[x][y] == 1) {
grid[x][y] = 0
neighborsOf(x, y).forEach {
markNeighbors(it.first, it.second)
}
}
}
This isn’t tail recursive because it doesn’t need to be (and would look more complicated). Essentially if the function is given a spot on the grid that’s a 1
, make it a 0
and call itself with all the valid neighbors. Technically we’re evaluating a lot of spots twice, but given the relatively small size of the grid I’m not going to optimize this further.
To solve part two, we execute the rest of our plan, which involves iteration over all the spots in the grid, maintaining the counter, and calling our function:
fun solvePart2(): Int {
var groups = 0
grid.forEachIndexed { x, row ->
row.forEachIndexed { y, spot ->
if (spot == 1) {
groups += 1
markNeighbors(x, y)
}
}
}
return groups
}
And that’s our second star of the day!
I hope you’ve learned something and as always, feedback is welcome!
14 days and we’ve earned 28 stars with 22 left to go!
Further Reading
- Index of All Solutions - All solutions for 2017, in Kotlin.
- My Github repo - Solutions and tests for each day.
- Solution - Full code for day 14.
- Advent of Code - Come join in and do these challenges yourself!
- Music to Code By - Subdivisions, by Rush.