src/Day01.kt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

fun main() {
    fun part1(input: List<String>): Int =
        input
            .map { it.toInt() }
            .zipWithNext()
            .count { it.second > it.first }

    fun part2(input: List<String>): Int =
        input
            .asSequence()
            .map { it.toInt() }
            .windowed(3)
            .map { it.sum() }
            .zipWithNext()
            .count { it.second > it.first }

    val testInput = readInput("Day01_test")
    check(part1(testInput) == 7)
    check(part2(testInput) == 5)

    val input = readInput("Day01")
    println(part1(input))
    println(part2(input))
}