advent-of-code/2021
advent-of-code/2021
1
0
Fork 0
Code Activity

day(14): refactor

Browse source 
do not keep separate functions for the same functionality
with different count of steps

Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
Matej Focko 2021-12-14 14:48:33 +01:00
parent bd4b174df9
commit 01bf675081
1 changed files with 6 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
src/year2021/day14/Day14.kt
View file

@ -55,21 +55,10 @@ fun String.toPolymerTemplate(): PolymerTemplate =
template + (it to template.getOrDefault(it, 0) + 1) template + (it to template.getOrDefault(it, 0) + 1)
} }
fun part1(input: List<String>): Long { fun run(input: List<String>, steps: Int): Long {
val polymer = Polymer(input.drop(2).toPolymerRules(), input.first()) val polymer = Polymer(input.drop(2).toPolymerRules(), input.first())
repeat(10) { repeat(steps) {
polymer.step()
}
val elements = polymer.elements
return elements.maxOf { it.value } - elements.minOf { it.value }
}
fun part2(input: List<String>): Long {
val polymer = Polymer(input.drop(2).toPolymerRules(), input.first())
repeat(40) {
polymer.step() polymer.step()
} }
@ -81,9 +70,9 @@ fun main() {
val sample = readInput(14, "sample") val sample = readInput(14, "sample")
val input = readInput(14, "input") val input = readInput(14, "input")
check(part1(sample) == 1588.toLong()) check(run(sample, 10) == 1588.toLong())
println(part1(input)) println(run(input, 10))
check(part2(sample) == 2188189693529) check(run(sample, 40) == 2188189693529)
println(part2(input)) println(run(input, 40))
} }