package de.skyrising.aoc2023.day4

import de.skyrising.aoc.*

val test = TestInput("""
    Card 1: 41 48 83 86 17 | 83 86  6 31 17  9 48 53
    Card 2: 13 32 20 16 61 | 61 30 68 82 17 32 24 19
    Card 3:  1 21 53 59 44 | 69 82 63 72 16 21 14  1
    Card 4: 41 92 73 84 69 | 59 84 76 51 58  5 54 83
    Card 5: 87 83 26 28 32 | 88 30 70 12 93 22 82 36
    Card 6: 31 18 13 56 72 | 74 77 10 23 35 67 36 11
""")

fun parse(input: PuzzleInput) = input.lines.map { line ->
    val (winning, have) = line.substringAfter(": ").split(" | ")
    have.ints().intersect(winning.ints()).size
}

@PuzzleName("")
fun PuzzleInput.part1() = parse(this).sumOf { if (it == 0) 0 else (1 shl (it - 1)) }

fun PuzzleInput.part2(): Any {
    val wins = parse(this)
    val deck = IntArray(wins.size) { 1 }
    deck.forEachIndexed { i, count ->
        for (j in i+1..i+wins[i]) deck[j] += count
    }
    return deck.sum()
}
