Sven/AoC
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor

Browse source
This commit is contained in:
Sven Weidauer 2024-12-11 19:35:56 +01:00
parent db704b8f36
commit 10b625d232
1 changed files with 30 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
2024/src/main/kotlin/day9.kt
View file

@ -5,36 +5,29 @@ sealed class Span {
data class File(val id: Int, override val length: Int) : Span() data class File(val id: Int, override val length: Int) : Span()
} }
fun main() { fun main() {
var currentFileId = 0 var currentFileId = 0
val disk = readInputString("day9.txt").mapIndexed { index, c -> val input = readInputString("day9.txt").mapIndexed { index, c ->
if (index % 2 == 0) { if (index % 2 == 0) {
Span.File(currentFileId++, c.digitToInt()) Span.File(currentFileId++, c.digitToInt())
} else { } else {
Span.Empty(c.digitToInt()) Span.Empty(c.digitToInt())
} }
}.toMutableList()
fun take(blocks: Int): Span.File {
var block: Span
do {
block = disk.removeLast()
} while (block !is Span.File)
if (block.length > blocks) {
disk.add(Span.File(block.id, block.length - blocks))
return Span.File(block.id, blocks)
} }
return block println("Part 1: ${part1(input)}")
} }
private fun part1(input: List<Span>): Long {
val disk = input.toMutableList()
var index = 1 var index = 1
while (index < disk.count()) { while (index < disk.count()) {
require(disk[index] is Span.Empty) require(disk[index] is Span.Empty)
val freeSpace = disk[index].length val freeSpace = disk[index].length
val fillData = take(freeSpace) val fillData = disk.takeFileBlocksFromEnd(freeSpace)
disk[index] = fillData disk[index] = fillData
if (fillData.length < freeSpace) { if (fillData.length < freeSpace) {
@ -45,15 +38,30 @@ fun main() {
} }
} }
val checkSum = calculateCheckSum(disk) val checkSum = disk.calculateCheckSum()
println("Part 1: $checkSum") return checkSum
} }
private fun calculateCheckSum(disk: List<Span>): Long { private fun MutableList<Span>.takeFileBlocksFromEnd(blocks: Int): Span.File {
var block: Span
do {
block = removeLast()
} while (block !is Span.File)
if (block.length > blocks) {
add(Span.File(block.id, block.length - blocks))
return Span.File(block.id, blocks)
}
return block
}
private fun List<Span>.calculateCheckSum(): Long {
var checkSum = 0L var checkSum = 0L
var currentBlock = 0L var currentBlock = 0L
for (file in disk) { for (file in this) {
require(file is Span.File) require(file is Span.File)
repeat(file.length) { repeat(file.length) {