Update Day 07

2024/day07.py

@@ -13,13 +13,48 @@ aocYear = identifyYear(sys.argv[0])
 
 path = getPath2Data(aocDay,aocYear)
 filename = getFilename2Data(aocDay)
-    
+
+def getData(data):
+    calcs=[]
+    for dataLine in data:
+        calc=[]
+        result, values = dataLine.split(':')
+        calc.append(int(result))
+        calc.append([int(value) for value in values.strip().split()])
+        calcs.append(calc)
+    return calcs
+
+def is_valid(result, values, part2):
+    if len(values) == 1:
+        return values[0]==result
+    if is_valid(result, [values[0]+values[1]] + values[2:], part2):
+        return True
+    if is_valid(result, [values[0]*values[1]] + values[2:], part2):
+        return True
+    if part2 and is_valid(result, [int(str(values[0])+str(values[1]))] + values[2:], part2):
+        return True
+    return False
+
+def bruteforce_calc(calc,part2):
+    ret = 0  
+    expectedResult = calc[0]
+    values = calc[1]
+    if is_valid(expectedResult, values, part2):
+        ret = expectedResult
+    return ret
+
 def taskA (data) -> int:
     gameScoreA = 0
+    calcs = getData(data)
+    for calc in calcs:
+        gameScoreA += bruteforce_calc(calc,False)
     return gameScoreA
 
 def taskB (data) -> int:
     gameScoreB = 0
+    calcs = getData(data)
+    for calc in calcs:
+        gameScoreB += bruteforce_calc(calc,True)
     return gameScoreB
 
 def task(task: int,data) -> int: