Day 13: Claw Contraption
Megathread guidelines
- Keep top level comments as only solutions, if you want to say something other than a solution put it in a new post. (replies to comments can be whatever)
- You can send code in code blocks by using three backticks, the code, and then three backticks or use something such as https://topaz.github.io/paste/ if you prefer sending it through a URL
FAQ
- What is this?: Here is a post with a large amount of details: https://programming.dev/post/6637268
- Where do I participate?: https://adventofcode.com/
- Is there a leaderboard for the community?: We have a programming.dev leaderboard with the info on how to join in this post: https://programming.dev/post/6631465
I have nothing. I hate Diophantine equations. That is all I have to say today.
(edit) I came back to it after a 24 hour break. All that nonsense about scoring multiple results really took me in yesterday.
import 'dart:math';
import 'package:collection/collection.dart';
List<List<Point<int>>> getMachines(List<String> lines) => lines
.splitBefore((e) => e == '')
.map((m) => m
.whereNot((e) => e.isEmpty)
.map((l) => RegExp(r'(\d+)')
.allMatches(l)
.map((m) => int.parse(m.group(0)!))
.toList())
.map((pr) => Point(pr.first, pr.last))
.toList())
.toList();
bool isInteger(num n) => (n - n.round()).abs() < 0.00001;
int cost(Point a, Point b, Point goal) {
var resA = (goal.x * b.y - goal.y * b.x) / (a.x * b.y - a.y * b.x);
var resB = (a.x * goal.y - a.y * goal.x) / (a.x * b.y - a.y * b.x);
return (isInteger(resA) && isInteger(resB))
? resA.round() * 3 + resB.round() * 1
: 0;
}
int solve(mx, p) => [for (var m in mx) cost(m[0], m[1], m[2] + p)].sum;
const large = Point(10000000000000, 10000000000000);
part1(List<String> lines) => solve(getMachines(lines), Point(0, 0));
part2(List<String> lines) => solve(getMachines(lines), large);
Well done! I like how youβre just looking for four integers instead of bothering with parsing the rest of the line.
Haskell
Whee, linear algebra! Converting between numeric types is a bit annoying in Haskell, but Iβm reasonably happy with this solution.
import Control.Monad
import Data.Matrix qualified as M
import Data.Maybe
import Data.Ratio
import Data.Vector qualified as V
import Text.Parsec
type C = (Int, Int)
readInput :: String -> [(C, C, C)]
readInput = either (error . show) id . parse (machine `sepBy` newline) ""
where
machine = (,,) <$> coords <*> coords <*> coords
coords =
(,)
<$> (manyTill anyChar (string ": X") >> anyChar >> num)
<*> (string ", Y" >> anyChar >> num)
<* newline
num = read <$> many1 digit
presses :: (C, C, C) -> Maybe C
presses ((ax, ay), (bx, by), (px, py)) =
do
let m = fromIntegral <$> M.fromLists [[ax, bx], [ay, by]]
m' <- either (const Nothing) Just $ M.inverse m
let [a, b] = M.toList $ m' * M.colVector (fromIntegral <$> V.fromList [px, py])
guard $ denominator a == 1
guard $ denominator b == 1
return (numerator a, numerator b)
main = do
input <- readInput <$> readFile "input13"
mapM_
(print . sum . map (\(a, b) -> 3 * a + b) . mapMaybe presses)
[ input,
map (\(a, b, (px, py)) -> (a, b, (10000000000000 + px, 10000000000000 + py))) input
]
Haskell
Pen and Paper solved these equations for me.
import Control.Arrow
import qualified Data.Char as Char
import qualified Data.List as List
import qualified Data.Maybe as Maybe
window6 :: [Int] -> [[Int]]
window6 [] = []
window6 is = List.splitAt 6
>>> second window6
>>> uncurry (:)
$ is
parse :: String -> [[Int]]
parse s = window6 . map read . words . List.filter ((Char.isDigit &&& Char.isSpace) >>> uncurry (||)) $ s
solveEquation (ax:ay:bx:by:tx:ty:[]) transformT
| (aNum `mod` aDenom) /= 0 = Nothing
| (bNum `mod` bDenom) /= 0 = Nothing
| otherwise = Just (abs $ aNum `div` aDenom, abs $ bNum `div` bDenom)
where
tx' = transformT tx
ty' = transformT ty
aNum = (bx*ty') - (by*tx')
aDenom = (ax*by) - (bx*ay)
bNum = (ax*ty') - (ay*tx')
bDenom = (ax*by) - (bx*ay)
part1 = map (flip solveEquation id)
>>> Maybe.catMaybes
>>> map (first (*3))
>>> map (uncurry (+))
>>> sum
part2 = map (flip solveEquation (+ 10000000000000))
>>> Maybe.catMaybes
>>> map (first (*3))
>>> map (uncurry (+))
>>> sum
main = getContents
>>= print
. (part1 &&& part2)
. parse
(Edit: coding style)
Python
Execution time: ~<1 millisecond (800 microseconds on my machine)
Good old school linear algebra from middle school. we can solve this really really fast. With minimal changes from part 1!
FastCode
from time import perf_counter_ns
import string
def profiler(method):
def wrapper_method(*args: any, **kwargs: any) -> any:
start_time = perf_counter_ns()
ret = method(*args, **kwargs)
stop_time = perf_counter_ns() - start_time
time_len = min(9, ((len(str(stop_time))-1)//3)*3)
time_conversion = {9: 'seconds', 6: 'milliseconds', 3: 'microseconds', 0: 'nanoseconds'}
print(f"Method {method.__name__} took : {stop_time / (10**time_len)} {time_conversion[time_len]}")
return ret
return wrapper_method
@profiler
def main(input_data):
part1_total_cost = 0
part2_total_cost = 0
for machine in input_data:
Ax,Ay,Bx,By,Px,Py = [ int(l[2:]) for l in machine.split() if l[-1] in string.digits ]
y,r = divmod((Ay * Px - Ax * Py), (Ay * Bx - Ax * By))
if r == 0:
x,r = divmod(Px - Bx * y, Ax)
if r == 0:
part1_total_cost += 3*x + y
y,r = divmod((Ay * (Px+10000000000000) - Ax * (Py+10000000000000)), (Ay * Bx - Ax * By))
if r == 0:
x,r = divmod((Px+10000000000000) - Bx * y, Ax)
if r == 0:
part2_total_cost += 3*x + y
return part1_total_cost,part2_total_cost
if __name__ == "__main__":
with open('input', 'r') as f:
input_data = f.read().strip().replace(',', '').split('\n\n')
part_one, part_two = main(input_data)
print(f"Part 1: {part_one}\nPart 2: {part_two}")
Itβs interesting that youβre not checking if the solution to x is a whole number. I guess the data doesnβt contain any counterexamples.
we are solving for y first. If there is a y then x is found easily.
(Ax)*x + (Bx)*y = Px
and (Ay)*x + (By)*y = Py
Because of Ax or Ay and Bx or By, lets pretend that they are not (A*x)*x
and (A*y)*y
for both. they are just names. could be rewritten as: (Aleft)*x + (Bleft)*y = Pleft
and (Aright)*x + (Bright)*y = Pright
but I will keep them short. solving for y turns into this:
y = (Ay*Px - Ax*Py) / (Ay*Bx - Ax*By)
if mod of 1 is equal to 0 then there is a solution. We can be confident that x is also a solution, too. Could there be an edge case? I didnβt proof it, but it works flawlessly for my solution.
Thankfully, divmod does both division and mod of 1 at the same time.
Thank you so much for your explanation! I kind of found some clues looking up perp dot products & other vector math things, but this breaks it down very nicely.
I implemented your solution in rust, and part 2 failed by +447,761,194,259 (this was using signed 64-bit integers, i64
). When I changed it to use signed 64 bit floating-point f64
and checked that the solution for x
produces a whole number it worked.
They do, if the remainder returned by divmod(β¦) wasnβt zero then it wouldnβt be divisble
This is a really excellent, clean solution! Would you mind breaking down how the piece of linear algebra works (for a shmo like me who doesnβt remember that stuff frum school heh π )
https://lemmy.world/comment/13950499
take the two equations, solve for y, and make sure y is fully divisible.
Haskell, 14 ms. The hardest part was the parser today. I somehow thought that the buttons could have negative values in X or Y too, so itβs a bit overcomplicated.
import Text.ParserCombinators.ReadP
int, signedInt :: ReadP Int
int = read <$> (many1 $ choice $ map char ['0' .. '9'])
signedInt = ($) <$> choice [id <$ char '+', negate <$ char '-'] <*> int
machine :: ReadP ((Int, Int), (Int, Int), (Int, Int))
machine = do
string "Button A: X"
xa <- signedInt
string ", Y"
ya <- signedInt
string "\nButton B: X"
xb <- signedInt
string ", Y"
yb <- signedInt
string "\nPrize: X="
x0 <- int
string ", Y="
y0 <- int
return ((xa, ya), (xb, yb), (x0, y0))
machines :: ReadP [((Int, Int), (Int, Int), (Int, Int))]
machines = sepBy machine (string "\n\n")
calc :: ((Int, Int), (Int, Int), (Int, Int)) -> Maybe (Int, Int)
calc ((ax, ay), (bx, by), (x0, y0)) = case
( (x0 * by - y0 * bx) `divMod` (ax * by - ay * bx)
, (x0 * ay - y0 * ax) `divMod` (bx * ay - by * ax)
) of
((a, 0), (b, 0)) -> Just (a, b)
_ -> Nothing
enlarge :: (a, b, (Int, Int)) -> (a, b, (Int, Int))
enlarge (u, v, (x0, y0)) = (u, v, (10000000000000 + x0, 10000000000000 + y0))
solve :: [((Int, Int), (Int, Int), (Int, Int))] -> Int
solve ts = sum
[ 3 * a + b
| Just (a, b) <- map calc ts
]
main :: IO ()
main = do
ts <- fst . last . readP_to_S machines <$> getContents
mapM_ (print . solve) [ts, map enlarge ts]
I wasted hours on the parsing, because my regex ([0-9]*)
was giving me empty strings. Made me feel very dumb when I worked it out