dnutiu/aoc-2023
dnutiu/aoc-2023
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

🎄Advent of Code - Day 4 ✨ part 2

Browse source
This commit is contained in:
Denis-Cosmin Nutiu 2023-12-04 18:24:13 +02:00
parent a7782e4b73
commit 12b957a64d
2 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/main/kotlin/AdventOfCode2023/day4/Scratchcards.kt
View file

@ -3,7 +3,7 @@ package AdventOfCode2023.day4
import AdventOfCode.Puzzle import AdventOfCode.Puzzle
import kotlin.collections.ArrayDeque import kotlin.collections.ArrayDeque
data class Card(val number: Int, val data: String) data class Card(val number: Int, val wonCards: Int)
class Scratchcards : Puzzle("2023", "4") { class Scratchcards : Puzzle("2023", "4") {
override fun partOne() { override fun partOne() {
@ -26,14 +26,13 @@ class Scratchcards : Puzzle("2023", "4") {
override fun partTwo() { override fun partTwo() {
var cardsTotal = 0 var cardsTotal = 0
val cardsList = inputData.mapIndexed { index, s -> Card(index + 1, s) }.toList() val cardsList = inputData.mapIndexed { index, s -> Card(index + 1, getWinningNumbersForCard(s)) }.toList()
val stack = ArrayDeque(cardsList) val stack = ArrayDeque(cardsList)
while (!stack.isEmpty()) { while (!stack.isEmpty()) {
val card = stack.removeFirst() val card = stack.removeFirst()
val wonCards = getWinningNumbersForCard(card) if (card.wonCards > 0) {
if (wonCards > 0) { val wonCardsList = cardsList.slice(card.number..<card.number+card.wonCards)
val wonCardsList = cardsList.slice(card.number..<card.number+wonCards)
wonCardsList.forEach { stack.addLast(it) } wonCardsList.forEach { stack.addLast(it) }
} }
cardsTotal += 1 cardsTotal += 1
@ -42,9 +41,9 @@ class Scratchcards : Puzzle("2023", "4") {
} }
private fun getWinningNumbersForCard( private fun getWinningNumbersForCard(
card: Card, card: String,
): Int { ): Int {
val parts = card.data.split("|") val parts = card.split("|")
val cleanParts = parts[1].split(Regex("\\s")).filter { it != "" }.toSet() val cleanParts = parts[1].split(Regex("\\s")).filter { it != "" }.toSet()
val winningNumbers = parts[0].split(":")[1].split(Regex("\\s")).filter { it != "" }.toSet() val winningNumbers = parts[0].split(":")[1].split(Regex("\\s")).filter { it != "" }.toSet()
return winningNumbers.filter { number -> cleanParts.contains(number) }.count() return winningNumbers.filter { number -> cleanParts.contains(number) }.count()

6
src/main/kotlin/Main.kt
View file

@ -1,7 +1,11 @@
import AdventOfCode2023.day4.Scratchcards import AdventOfCode2023.day4.Scratchcards
import kotlin.system.measureTimeMillis
fun main(args: Array<String>) { fun main(args: Array<String>) {
val t = Scratchcards() val t = Scratchcards()
// t.partOne() // t.partOne()
t.partTwo() val time = measureTimeMillis {
t.partTwo()
}
println("Took $time ms.")
} }