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

Solve AOC day 7 pt 2

Browse Source
This commit is contained in:
Nat Quayle Nelson
2021-12-07 11:20:50 -07:00
parent 97e2369fcb
commit 505e7fd054
2 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -31,7 +31,8 @@
(load "day7.kiss")
(assert (= 2 (bestPosToAlignAt (crabPositions "src/year2021/inputs/day7-example.txt"))))
(let [positions (crabPositions "src/year2021/inputs/day7.txt")]
(print (fuelToAlignAt (bestPosToAlignAt positions) positions))))
(assert (= 356958 (fuelToAlignAt (bestPosToAlignAt positions) positions)))
(assert (= 105461913 (fuelToAlignAt (bestPosToAlignAt positions true) positions true)))))
(dayTodo 8)
(dayTodo 9)
(dayTodo 10)

19
projects/aoc/src/year2021/day7.kiss
View File

@@ -1,17 +1,24 @@
(function crabPositions [file]
(map (.split (first (Util.readLines file)) ",") Std.parseInt))
(function :Int fuelToAlignAt [x :Array<Int> positions]
(apply + (for crabX positions (Math.abs (- x crabX)))))
(function :Int summation [n]
(/ (* n (+ 1 n)) 2))
(function :Int bestPosToAlignAt [:Array<Int> positions]
(function :Int fuelToAlignAt [x :Array<Int> positions &opt :Bool part2]
(apply + (for crabX positions
(let [distance (Math.abs (- x crabX))]
(if part2
(summation distance)
distance)))))
(function :Int bestPosToAlignAt [:Array<Int> positions &opt :Bool part2]
(let [&mut minX (apply min positions)
&mut maxX (apply max positions)]
(loop
(let [middleX (Math.floor (/ (+ minX maxX) 2))
middleFuel (fuelToAlignAt middleX positions)
leftBy1Fuel (fuelToAlignAt (- middleX 1) positions)
rightBy1Fuel (fuelToAlignAt (+ middleX 1) positions)]
middleFuel (fuelToAlignAt middleX positions ?part2)
leftBy1Fuel (fuelToAlignAt (- middleX 1) positions ?part2)
rightBy1Fuel (fuelToAlignAt (+ middleX 1) positions ?part2)]
~[minX middleX maxX]
(cond
((= minX middleX maxX)