42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
# SPDX-License-Identifier: MIT
|
|
# Copyright (c) 2022 Akumatic
|
|
#
|
|
# https://adventofcode.com/2022/day/8
|
|
|
|
def read_file(filename: str = "input.txt") -> list:
|
|
with open(f"{__file__.rstrip('code.py')}{filename}", "r") as f:
|
|
lines = [[int(x) for x in line.strip()] for line in f.readlines()]
|
|
return lines
|
|
|
|
def calculate_scenic_score(field: list, i: int, j: int) -> int:
|
|
dirs = {"u": (0, -1), "d": (0, 1), "l": (1, 0), "r": (-1, 0)}
|
|
count = {"u": 0, "d": 0, "l": 0, "r": 0}
|
|
for dir, val in dirs.items():
|
|
x, y = i, j
|
|
while x + val[0] >= 0 and x + val[0] < len(field[0]) and \
|
|
y + val[1] >= 0 and y + val[1] < len(field):
|
|
x += val[0]
|
|
y += val[1]
|
|
count[dir] += 1
|
|
if field[y][x] >= field[j][i]:
|
|
break
|
|
return count["u"] * count["d"] * count["l"] * count["r"]
|
|
|
|
def part1(vals: list) -> int:
|
|
return sum([any((
|
|
not any(vals[j][x] >= vals[j][i] for x in range(0, i)), # left
|
|
not any(vals[j][x] >= vals[j][i] for x in range(i+1, len(vals[0]))), # right
|
|
not any(vals[y][i] >= vals[j][i] for y in range(0, j)), # down
|
|
not any(vals[y][i] >= vals[j][i] for y in range(j+1, len(vals))) # up
|
|
)) for j in range(len(vals)) for i in range(len(vals[0]))])
|
|
|
|
def part2(vals: list) -> int:
|
|
scenic_scores = [calculate_scenic_score(vals, i, j) for
|
|
j in range(len(vals)) for i in range(len(vals[0]))]
|
|
return max(scenic_scores)
|
|
|
|
if __name__ == "__main__":
|
|
vals = read_file()
|
|
print(f"Part 1: {part1(vals)}")
|
|
print(f"Part 2: {part2(vals)}")
|