diff --git a/src/aoc_2024/day4/part1.py b/src/aoc_2024/day4/part1.py index 8010888..69776e3 100644 --- a/src/aoc_2024/day4/part1.py +++ b/src/aoc_2024/day4/part1.py @@ -106,22 +106,18 @@ def find_xmas(data: np.array): for i in range(data.shape[0]): for j in range(data.shape[1]): if data[i, j] == "X": - if is_horizontal_forwards(data, i, j): - score += 1 - if is_horizontal_backwards(data, i, j): - score += 1 - if is_vertical_downwards(data, i, j): - score += 1 - if is_vertical_upwards(data, i, j): - score += 1 - if is_diagonal_forwards_downwards(data, i, j): - score += 1 - if is_diagonal_forwards_upwards(data, i, j): - score += 1 - if is_diagonal_backwards_downwards(data, i, j): - score += 1 - if is_diagonal_backwards_upwards(data, i, j): - score += 1 + score += sum( + [ + is_horizontal_forwards(data, i, j), + is_horizontal_backwards(data, i, j), + is_vertical_downwards(data, i, j), + is_vertical_upwards(data, i, j), + is_diagonal_forwards_downwards(data, i, j), + is_diagonal_forwards_upwards(data, i, j), + is_diagonal_backwards_downwards(data, i, j), + is_diagonal_backwards_upwards(data, i, j), + ] + ) return score