day 4 - part 1 - refactoring

This commit is contained in:
Stanislas Jouffroy 2024-12-04 15:55:47 +01:00
parent 38013bf705
commit 93318458bf

View file

@ -106,22 +106,18 @@ def find_xmas(data: np.array):
for i in range(data.shape[0]): for i in range(data.shape[0]):
for j in range(data.shape[1]): for j in range(data.shape[1]):
if data[i, j] == "X": if data[i, j] == "X":
if is_horizontal_forwards(data, i, j): score += sum(
score += 1 [
if is_horizontal_backwards(data, i, j): is_horizontal_forwards(data, i, j),
score += 1 is_horizontal_backwards(data, i, j),
if is_vertical_downwards(data, i, j): is_vertical_downwards(data, i, j),
score += 1 is_vertical_upwards(data, i, j),
if is_vertical_upwards(data, i, j): is_diagonal_forwards_downwards(data, i, j),
score += 1 is_diagonal_forwards_upwards(data, i, j),
if is_diagonal_forwards_downwards(data, i, j): is_diagonal_backwards_downwards(data, i, j),
score += 1 is_diagonal_backwards_upwards(data, i, j),
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
return score return score