kiss-lang/kiss-vscode
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

actually solve aoc day 6 pt 2

Browse Source
This commit is contained in:
Nat Quayle Nelson
2021-12-06 13:42:37 -07:00
parent afa3699cfe
commit 6ecb87c5ae
3 changed files with 23 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
projects/aoc/src/year2021/Solutions2021.hx
View File

@@ -3,6 +3,8 @@ package year2021;
import kiss.Prelude;
import haxe.Constraints;
import haxe.ds.Option;
import haxe.Int64;
import haxe.Int64Helper;
#if day4
import year2021.Day4;
#end

5
projects/aoc/src/year2021/Solutions2021.kiss
View File

@@ -20,7 +20,10 @@
(dayTodo 5)
(day 6
(load "day6.kiss")
(print (countAfter "src/year2021/inputs/day6.txt" 80)))
(assert (= "5934" (countAfter "src/year2021/inputs/day6-example.txt" 80)))
(assert (= "26984457539" (countAfter "src/year2021/inputs/day6-example.txt" 256)))
(assert (= "346063" (countAfter "src/year2021/inputs/day6.txt" 80)))
(assert (= "1572358335990" (countAfter "src/year2021/inputs/day6.txt" 256))))
(dayTodo 7)
(dayTodo 8)
(dayTodo 9)

32
projects/aoc/src/year2021/day6.kiss
View File

@@ -1,28 +1,30 @@
(defMacro dictInc [theMap key &builder b amount]
`(let [theMap ,theMap]
(unless (dictGet theMap ,key)
(dictSet theMap ,key 0))
(+= (dictGet theMap ,key) ,amount)))
(defMacro dictInc [theMap key amount]
`(let [theMap ,theMap key ,key amount ,amount &mut count (dictGet theMap key)]
(unless count
(dictSet theMap key (set count (Int64Helper.fromFloat 0))))
(dictSet theMap key #|count + amount|#)))
// Store the simulation state in a map of age => count.
(function :Map<Int,Int> ageMap [:String file]
(function :Map<Int,Int64> ageMap [:String file]
(let [:Array<Int> list (map (.split (first (Util.readLines file)) ",") Std.parseInt)
:Map<Int,Int> theMap (new Map)]
:Map<Int,Int64> theMap (new Map)]
// special key -1 corresponds to total count
(dictSet theMap -1 list.length)
(dictSet theMap -1 (Int64Helper.fromFloat list.length))
(doFor age (range 9)
(dictSet theMap age 0))
(dictSet theMap age (Int64Helper.fromFloat 0)))
(doFor age list
(+= (dictGet theMap age) 1))
(let [count (dictGet theMap age)
one (Int64Helper.fromFloat 1)]
(dictSet theMap age #|count + one|#)))
theMap))
(function :Map<Int,Int> stepAgeMap [:Map<Int,Int> theMap]
(function :Map<Int,Int64> stepAgeMap [:Map<Int,Int64> theMap]
// Lanternfish are guaranteed to be between 0-8 years old
(let [:Map<Int,Int> newMap (new Map)]
(let [:Map<Int,Int64> newMap (new Map)]
(dictSet newMap -1 (dictGet theMap -1))
(doFor age (range 9)
(let [count (dictGet theMap age)]
(case age
(case ~age
// Lanternfish giving birth:
(0
(dictInc newMap 8 count)
@@ -32,8 +34,8 @@
(dictInc newMap (- age 1) count)))))
newMap))
(function countAfter [file days]
(function :String countAfter [file days]
(let [&mut ageMap (ageMap file)]
(doFor i (range days)
(set ageMap (stepAgeMap ageMap)))
(nth ageMap -1)))
(Int64.toStr (dictGet ageMap -1))))