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

day(05): fix DRY

Browse source 
Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
Matej Focko 2021-12-05 11:26:31 +01:00
parent 0492a86f08
commit 67a18e2687
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/day05/Day05.kt
View file

@ -47,22 +47,11 @@ fun readVector(input: String): Vector {
return Vector(readPoint(from), readPoint(to)) return Vector(readPoint(from), readPoint(to))
} }
fun part1(input: List<Vector>): Int { fun findOverlaps(input: List<Vector>): Int {
var points: MutableMap<Point, Int> = mutableMapOf() var points: MutableMap<Point, Int> = mutableMapOf()
input
input.filter { it.horizontalOrVertical }
.flatMap { it.points } .flatMap { it.points }
.forEach { points[it] = 1 + points.getOrDefault(it, 0) } .forEach { points[it] = 1 + points.getOrDefault(it, 0) }
return points.count { (_, count) -> count >= 2 }
}
fun part2(input: List<Vector>): Int {
var points: MutableMap<Point, Int> = mutableMapOf()
input.flatMap { it.points }
.forEach { points[it] = 1 + points.getOrDefault(it, 0) }
return points.count { (_, count) -> count >= 2 } return points.count { (_, count) -> count >= 2 }
} }
@ -70,9 +59,9 @@ fun main() {
val testInput = readInput(5, "test_input").map { readVector(it) } val testInput = readInput(5, "test_input").map { readVector(it) }
val input = readInput(5, "input").map { readVector(it) } val input = readInput(5, "input").map { readVector(it) }
check(part1(testInput) == 5) check(findOverlaps(testInput.filter { it.horizontalOrVertical }) == 5)
println(part1(input)) println(findOverlaps(input.filter { it.horizontalOrVertical }))
check(part2(testInput) == 12) check(findOverlaps(testInput) == 12)
println(part2(input)) println(findOverlaps(input))
} }