day 3 slight simplification

main
magical 2023-12-04 01:29:42 +00:00
parent 46ba7f1593
commit 6c0f46fd2b
1 changed files with 6 additions and 15 deletions

View File

@ -38,19 +38,18 @@ for no, line in enumerate(data):
i = j
else:
i += 1
for n, pos in zip(digits, places):
if any(nexttosymbol(no, i) for i in pos):
print(n)
part1 += n
seen = set()
nearby_gears = set()
for i in pos:
for g in findgears(no,i):
if g not in seen:
gears.setdefault(g, []).append(n)
seen.add(g)
nearby_gears.update(findgears(no, i))
for g in nearby_gears:
gears.setdefault(g, []).append(n)
print(part1)
part2 = 0
@ -58,11 +57,3 @@ for g, nums in gears.items():
if len(nums) == 2:
part2 += nums[0] * nums[1]
print(part2)