diff --git a/day03.py b/day03.py index 640aad3..c72b117 100644 --- a/day03.py +++ b/day03.py @@ -24,7 +24,7 @@ class Day03(AocSameParser[list[Number]], Aoc2[list[Number], list[Number]]): symbol = None for x in range(max(0, startx - 1), min(len(inpt[0]), endx + 1)): yrange = [liney - 1, liney, liney + 1] - if x != startx - 1 and x != endx: + if x not in (endx, startx - 1): yrange.remove(liney) if liney == 0: yrange.remove(liney - 1) @@ -76,7 +76,7 @@ class Day03(AocSameParser[list[Number]], Aoc2[list[Number], list[Number]]): return numbers def part1(self, inpt: list[Number]) -> int: - return sum([number.value for number in inpt if number.symbol is not None]) + return sum(number.value for number in inpt if number.symbol is not None) def part2(self, inpt: list[Number]) -> Any: gears_to_number = defaultdict(list)