-
Notifications
You must be signed in to change notification settings - Fork 0
/
day_11_2D_arrays.py
35 lines (31 loc) · 949 Bytes
/
day_11_2D_arrays.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
def find_max_hourglass(arr):
height = 3
width = 3
# result = [] option
result = -64 # Due to constraints
# iterate through rows of arr
for i in range(len(arr)):
if i > height:
break
# iterate through columns of arr
for j in range(len(arr[0])):
if j > width:
break
sum = 0
# iterate through rows and columns of the hourglasses
for k in range(j, j+3):
sum += arr[i][k]
sum += arr[i+2][k]
sum += arr[i + 1][j + 1]
# result.append(sum) option
if result < sum:
result = sum
# return max(result) option
return result
def main():
arr = []
for _ in range(6):
arr.append(list(map(int, input().rstrip().split())))
print(find_max_hourglass(arr))
if __name__ == "__main__":
main()