diff options
Diffstat (limited to '2024/day4/solve.py')
| -rw-r--r-- | 2024/day4/solve.py | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/2024/day4/solve.py b/2024/day4/solve.py new file mode 100644 index 0000000..da301c9 --- /dev/null +++ b/2024/day4/solve.py @@ -0,0 +1,43 @@ +def check_direction(lines: list[str], row_idx: int, col_idx: int, row_step: int, column_step: int) -> bool: + column_count = len(lines[0]) + row_count = len(lines) + for step, letter in enumerate("MAS"): + current_row = row_idx + (step + 1) * row_step + current_col = col_idx + (step + 1) * column_step + if current_row >= row_count or current_row < 0: + return False + + if current_col >= column_count or current_col < 0: + return False + + if lines[current_row][current_col] != letter: + return False + + return True + +def get_current_pos_words(lines: list[str], row_idx: int, col_idx: int) -> int: + if lines[row_idx][col_idx] != 'X': + return 0 + + count = 0 + for row_step in range(-1, 2): + for column_step in range(-1, 2): + if row_step == 0 and column_step == 0: + continue + + if check_direction(lines, row_idx, col_idx, row_step, column_step): + count += 1 + + return count + +data = [] +with open("input") as f: + for line in f: + data.append(line.strip()) + +result = 0 +for row, line in enumerate(data): + for col in range(len(line)): + result += get_current_pos_words(data, row, col) + +print(result)
\ No newline at end of file |