Spaces:
Running
Running
File size: 2,237 Bytes
a4da721 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 |
def parse_data(file):
with open(file) as f:
raw_data = f.readlines()
# Read and parse the data
data = []
for line in raw_data:
line = line.strip("\n")
target, nums = line.split(":")
target = int(target)
nums = [int(num) for num in nums.split()]
data.append((target, nums))
return data
def compute(a, b, op):
"""Compute given the op"""
if op == "+":
return a + b
if op == "*":
return a * b
def check_seq(nums, target, seq):
total = nums[0]
for i in range(len(seq)):
b = nums[i+1]
op = seq[i]
total = compute(total, b, op)
if total > target:
# Dead-end
return -1
# Check that we equal target and use all nums
return total == target and len(seq) == len(nums) - 1
def bfs(target, nums, ops):
q = ops.copy()
dead_ends = set()
while len(q) > 0:
# print(q)
# print(dead_ends)
seq = q.pop(0)
if seq in dead_ends:
break
check = check_seq(nums, target, seq)
# print(nums, target, seq, check)
if check == -1:
dead_ends.add(seq)
continue
elif check == True:
return True
else:
if len(seq) < len(nums)-1:
for op in ops:
q.append(seq+op)
return False
data = parse_data(file="input.txt")
ops = ["+", "*"]
total = 0
for target, nums in data:
result = bfs(target, nums, ops)
# print("*"*20)
# print(target, nums)
# print("Result:", result)
# print("*"*20)
if result:
total += target
print(total)
## Part 2
def compute(a, b, op):
"""Compute given the op"""
if op == "+":
return a + b
elif op == "*":
return a * b
elif op == "|":
return int(str(a) + str(b))
else:
raise ValueError(f"Op {op} Unknown")
ops = ["+", "*", "|"]
data = parse_data(file="input.txt")
total = 0
for target, nums in data:
result = bfs(target, nums, ops)
# print("*"*20)
# print(target, nums)
# print("Result:", result)
# print("*"*20)
if result:
total += target
print(total)
|