c4patino/advent-of-code
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Made some slightly smarter merge blocks code

Browse source
This commit is contained in:
Ceferino Patino 2024-12-09 10:37:57 -06:00
commit 69a08312e8
Signed by: c4patino
SSH key fingerprint: SHA256:Wu+cU1t+7zVT9wzew/4meNmeulo66NzMqc22MdDBgXI
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
2024/day9/day9.go
View file

@ -33,19 +33,22 @@ func calculateChecksum(blocks []Block) int {
} }
func mergeBlocks(blocks []Block) []Block { func mergeBlocks(blocks []Block) []Block {
merged := []Block{} if len(blocks) == 0 {
return blocks
}
merged := blocks[:0]
for _, block := range blocks { for _, block := range blocks {
if block.id != nil { if block.id != nil {
merged = append(merged, block) merged = append(merged, block)
continue continue
} }
if merged[len(merged)-1].id == nil { if len(merged) > 0 && merged[len(merged)-1].id == nil {
merged[len(merged)-1].size += block.size merged[len(merged)-1].size += block.size
continue } else {
merged = append(merged, block)
} }
merged = append(merged, block)
} }
return merged return merged