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 i = j
else: else:
i += 1 i += 1
for n, pos in zip(digits, places): for n, pos in zip(digits, places):
if any(nexttosymbol(no, i) for i in pos): if any(nexttosymbol(no, i) for i in pos):
print(n) print(n)
part1 += n part1 += n
seen = set() nearby_gears = set()
for i in pos: for i in pos:
for g in findgears(no,i): nearby_gears.update(findgears(no, i))
if g not in seen: for g in nearby_gears:
gears.setdefault(g, []).append(n) gears.setdefault(g, []).append(n)
seen.add(g)
print(part1) print(part1)
part2 = 0 part2 = 0
@ -58,11 +57,3 @@ for g, nums in gears.items():
if len(nums) == 2: if len(nums) == 2:
part2 += nums[0] * nums[1] part2 += nums[0] * nums[1]
print(part2) print(part2)