From b83d6446daab4f0a625765b226a37e3f779be7d2 Mon Sep 17 00:00:00 2001 From: mhsn Date: Sat, 7 Dec 2024 12:00:00 +0000 Subject: 2024-07 python p1,p2 --- 2024/07/python/main.py | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 2024/07/python/main.py diff --git a/2024/07/python/main.py b/2024/07/python/main.py new file mode 100644 index 0000000..0e72772 --- /dev/null +++ b/2024/07/python/main.py @@ -0,0 +1,35 @@ +from fileinput import input + +lines = [line.strip() for line in input()] + + +def solve(w, acc, rest, gold=False): + if not rest: + return w == acc + if w < acc: + return False + head, *rest = rest + return ( + solve(w, acc + head, rest, gold) + or solve(w, acc * head, rest, gold) + or (gold and solve(w, int(str(acc) + str(head)), rest, gold)) + ) + + +silver = 0 +gold = 0 + +for line in lines: + want, nums = line.split(":") + want = int(want) + nums = [int(n) for n in nums.split()] + head, *rest = nums + + if solve(want, head, rest): + silver += want + if solve(want, head, rest, True): + gold += want + + +print("silver:", silver) +print("gold:", gold) -- cgit v1.2.3