Day 7 part 2 bruteforced

This commit is contained in:
Daniel Flanagan 2021-12-07 10:09:52 -06:00
parent b9a249df6b
commit e88a868165
Signed by: lytedev
GPG key ID: 5B2020A0F9921EF4

View file

@ -1,6 +1,9 @@
import std/[strutils, sequtils, tables, strformat]
import ./common
proc f(n: int): int = toSeq(0..abs(n)).foldl(a + b, 0)
proc crabsDistance(input: string): int =
result = int.high()
var max_try = 0
@ -11,10 +14,22 @@ proc crabsDistance(input: string): int =
let fuel = dists.foldl(a + abs(b - t), 0)
if fuel < result: result = fuel
proc crabsBigDistance(input: string): int =
result = int.high()
var max_try = 0
let dists = input.split(',').map(parseInt)
for d in dists:
if d > max_try: max_try = d
for t in 0..max_try:
let fuel = dists.foldl(a + f(b - t), 0)
if fuel < result: result = fuel
let input = 7.loadInputText()
time("day 7 part 1"): echo input.crabsDistance()
time("day 7 part 2"): echo input.crabsBigDistance()
# time("day 7 part 2"): echo "?"
when not defined(release):
let testInput = "16,1,2,0,4,2,7,1,2,14"
doAssert testInput.crabsDistance() == 37
doAssert testInput.crabsBigDistance() == 168