66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
def is_valid_update(update, order_reqs):
|
||
|
|
valid_update = True
|
||
|
|
for i in range(0, len(update)):
|
||
|
|
if update[i] in order_reqs:
|
||
|
|
for j in range(0, len(update)):
|
||
|
|
if update[j] in order_reqs[update[i]] and j < i:
|
||
|
|
valid_update = False
|
||
|
|
|
||
|
|
return valid_update
|
||
|
|
|
||
|
|
def fix_update(update, order_reqs):
|
||
|
|
if is_valid_update(update, order_reqs):
|
||
|
|
return update
|
||
|
|
else:
|
||
|
|
fixed_update = update
|
||
|
|
for i in range(0, len(update)):
|
||
|
|
if update[i] in order_reqs:
|
||
|
|
for j in range(0, len(update)):
|
||
|
|
if update[j] in order_reqs[update[i]] and j < i:
|
||
|
|
fixed_update[i], fixed_update[j] = fixed_update[j], fixed_update[i]
|
||
|
|
if is_valid_update(fixed_update, order_reqs):
|
||
|
|
return fixed_update
|
||
|
|
else:
|
||
|
|
return fix_update(fixed_update, order_reqs)
|
||
|
|
|
||
|
|
with open("./2024/inputs/day05.txt") as f:
|
||
|
|
lines = [l.strip() for l in f.readlines()]
|
||
|
|
|
||
|
|
order_reqs = {}
|
||
|
|
updates = []
|
||
|
|
nums_ended = False
|
||
|
|
|
||
|
|
for l in lines:
|
||
|
|
if l == "":
|
||
|
|
nums_ended = True
|
||
|
|
continue
|
||
|
|
|
||
|
|
if not nums_ended:
|
||
|
|
before, after = [int(n) for n in l.split("|")]
|
||
|
|
if before not in order_reqs:
|
||
|
|
order_reqs[before] = [after]
|
||
|
|
else:
|
||
|
|
order_reqs[before].append(after)
|
||
|
|
|
||
|
|
else:
|
||
|
|
updates.append([int(n) for n in l.split(",")])
|
||
|
|
|
||
|
|
middle_total = 0
|
||
|
|
invalid_updates = []
|
||
|
|
for update in updates:
|
||
|
|
if is_valid_update(update, order_reqs):
|
||
|
|
middle_total += update[int((len(update) - 1) / 2)]
|
||
|
|
else:
|
||
|
|
invalid_updates.append(update)
|
||
|
|
|
||
|
|
print(middle_total)
|
||
|
|
|
||
|
|
middle_total_fixed = 0
|
||
|
|
for update in invalid_updates:
|
||
|
|
fixed_update = fix_update(update, order_reqs)
|
||
|
|
middle_total_fixed += fixed_update[int((len(fixed_update) - 1) / 2)]
|
||
|
|
|
||
|
|
print(middle_total_fixed)
|