Add bisect utility

This is too common of an optimization to not have this readily accessible. And I kinda like how this worked out, too. Go is fun.
This commit is contained in:
2022-06-21 00:12:06 -05:00
parent 23bf2e9d84
commit 94471df109
2 changed files with 35 additions and 20 deletions

View File

@ -104,26 +104,10 @@ func (d *Day14) Part1() string {
func (d *Day14) Part2() string {
oreAvailable := int64(1000000000000)
estimate := oreAvailable / d.getOreRequiredForFuel(1)
high := estimate * 2
low := estimate
lastSuccess := low
lastFailure := high
fuelProduced := low
for math.Abs(float64(lastFailure-lastSuccess)) > 1 {
oreConsumed := d.getOreRequiredForFuel(fuelProduced)
adjustment := (lastFailure - lastSuccess) / 2
if oreConsumed < oreAvailable {
lastSuccess = fuelProduced
} else {
lastFailure = fuelProduced
adjustment = -adjustment
}
fuelProduced += adjustment
}
lastSuccess := u.Bisect(estimate, estimate*2, 1, func(val int64) bool {
oreConsumed := d.getOreRequiredForFuel(val)
return oreConsumed < oreAvailable
})
return fmt.Sprintf("Maximum fuel we can make from 1 trillion ore: %s%d%s", u.TextBold, lastSuccess, u.TextReset)
}

31
utilities/bisect.go Normal file
View File

@ -0,0 +1,31 @@
package utilities
import "math"
// Bisect takes a known-good low and known-bad high value as the bounds
// to bisect, and a function to test each value for success or failure.
// If the function succeeds, the value is adjusted toward the maximum,
// and if the function fails, the value is adjusted toward the minimum.
// The final value is returned when the difference between the success
// and the failure is less than or equal to the acceptance threshold
// (usually 1, for integers).
func Bisect[T Number](low, high, threshold T, tryFunc func(val T) bool) T {
lastSuccess := low
lastFailure := high
currVal := low
for T(math.Abs(float64(lastFailure-lastSuccess))) > threshold {
success := tryFunc(currVal)
adjustment := (lastFailure - lastSuccess) / 2
if success {
lastSuccess = currVal
} else {
lastFailure = currVal
adjustment = -adjustment
}
currVal += adjustment
}
return currVal
}