/
day09.py
38 lines (31 loc) · 1.16 KB
/
day09.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
with open("day09-input.txt", "r") as f:
data = f.readlines()
data = [[int(y) for y in x.strip()] for x in data]
def sumOfTheRiskLevelsOfAllLowPoints(data):
sum = 0
for i in range(len(data)):
for j in range(len(data[i])):
v = data[i][j]
u = 999999999
d = 999999999
l = 999999999
r = 999999999
if i > 0:
u = data[i - 1][j]
if i < len(data) - 1:
d = data[i + 1][j]
if j > 0:
l = data[i][j - 1]
if j < len(data[i]) - 1:
r = data[i][j + 1]
if u > v and d > v and l > v and r > v:
sum += 1 + data[i][j]
return sum
def multiplyTogetherTheSizesOfTheThreeLargestBasins(data):
largest = [0, 0, 0]
return sum(largest)
if __name__ == "__main__":
# What is the sum of the risk levels of all low points on your heightmap?
print(f"Day09a: {sumOfTheRiskLevelsOfAllLowPoints(data)}") # 545
# What do you get if you multiply together the sizes of the three largest basins?
print(f"Day09b: {multiplyTogetherTheSizesOfTheThreeLargestBasins(data)}") # ???