Update main.py

This commit is contained in:
akp 2024-12-07 15:30:32 +00:00
parent 42d3c90d44
commit 4902fa69c4
No known key found for this signature in database
GPG key ID: CF8D58F3DEB20755

View file

@ -49,8 +49,11 @@ def one(instr: str):
cases = parse(instr) cases = parse(instr)
return reduce( return reduce(
lambda x, y: x + y, lambda x, y: x + y,
(target if solve(target, numbers, use_concat=False) else 0 for (target, numbers) in cases), (
0 target if solve(target, numbers, use_concat=False) else 0
for (target, numbers) in cases
),
0,
) )
@ -58,8 +61,11 @@ def two(instr: str):
cases = parse(instr) cases = parse(instr)
return reduce( return reduce(
lambda x, y: x + y, lambda x, y: x + y,
(target if solve(target, numbers, use_concat=True) else 0 for (target, numbers) in cases), (
0 target if solve(target, numbers, use_concat=True) else 0
for (target, numbers) in cases
),
0,
) )