diff options
Diffstat (limited to 'src/Day17.kt')
-rw-r--r-- | src/Day17.kt | 78 |
1 files changed, 40 insertions, 38 deletions
diff --git a/src/Day17.kt b/src/Day17.kt index 5eaf498..5e0170b 100644 --- a/src/Day17.kt +++ b/src/Day17.kt @@ -1,60 +1,62 @@ import kotlin.math.max import kotlin.math.sign -data class Target(val left: Int, val right: Int, val bottom: Int, val top: Int) - -class Probe(private var vel: Pos2D, private val target: Target) { - companion object { - fun search(target: Target): Pair<Int, Int> { - var highest = 0 - var count = 0 - - for (vx in 0..1000) { - for (vy in -1000..1000) { - val probe = Probe(Pos2D(vx, vy), target) - var currentHighest = 0 - - while (probe.canHitTarget) { - probe.step() - currentHighest = max(currentHighest, probe.pos.y) - - if (probe.isInTarget) { - count++ - highest = max(highest, currentHighest) - break +object Day17 { + data class Target(val left: Int, val right: Int, val bottom: Int, val top: Int) + + private class Probe(private var vel: Pos2D, private val target: Target) { + companion object { + fun search(target: Target): Pair<Int, Int> { + var highest = 0 + var count = 0 + + for (vx in 0..1000) { + for (vy in -1000..1000) { + val probe = Probe(Pos2D(vx, vy), target) + var currentHighest = 0 + + while (probe.canHitTarget) { + probe.step() + currentHighest = max(currentHighest, probe.pos.y) + + if (probe.isInTarget) { + count++ + highest = max(highest, currentHighest) + break + } } } } + + return highest to count } + } + + private var pos = Pos2D(0, 0) - return highest to count + private fun step() { + pos += vel + vel = vel.copy(x = vel.x - vel.x.sign, y = vel.y - 1) } - } - private var pos = Pos2D(0, 0) + private val canHitTarget + get() = pos.y > target.bottom - fun step() { - pos += vel - vel = vel.copy(x = vel.x - vel.x.sign, y = vel.y - 1) + private val isInTarget + get() = pos.x in target.left..target.right && pos.y in target.bottom..target.top } - val canHitTarget - get() = pos.y > target.bottom - - val isInTarget - get() = pos.x in target.left..target.right && pos.y in target.bottom..target.top + fun bothParts(input: Target) = Probe.search(input) } fun main() { - fun bothParts(input: Target): Pair<Int, Int> = Probe.search(input) - - val testInput = Target(20, 30, -10, -5) - val testOutput = bothParts(testInput) + val testInput = Day17.Target(20, 30, -10, -5) + val testOutput = Day17.bothParts(testInput) check(testOutput.first == 45) check(testOutput.second == 112) - val input = Target(192, 251, -89, -59) - val output = bothParts(input) + val input = Day17.Target(192, 251, -89, -59) + val output = Day17.bothParts(input) println(output.first) println(output.second) } |