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

@ -44,12 +44,11 @@ for no, line in enumerate(data):
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)
@ -58,11 +57,3 @@ for g, nums in gears.items():
if len(nums) == 2:
part2 += nums[0] * nums[1]
print(part2)