Untitled
unknown
plain_text
8 months ago
1.3 kB
9
Indexable
#!/bin/python3
import math
import os
import random
import re
import sys
#
# Complete the 'hourglassSum' function below.
#
# The function is expected to return an INTEGER.
# The function accepts 2D_INTEGER_ARRAY arr as parameter.
#
def hourglassSum(arr):
max_sum = -float('inf') # Initialize with the smallest possible value
for i in range(4): # We only loop till row 3, since hourglass needs 3 rows
for j in range(4): # We only loop till column 3, since hourglass needs 3 columns
# Calculate the sum of the current hourglass
hourglass = (arr[i][j] + arr[i][j+1] + arr[i][j+2] + # Top row
arr[i+1][j+1] + # Middle row (only middle element)
arr[i+2][j] + arr[i+2][j+1] + arr[i+2][j+2]) # Bottom row
# Update max_sum if the current hourglass sum is greater
max_sum = max(max_sum, hourglass)
return max_sum
if __name__ == '__main__':
fptr = open(os.environ['OUTPUT_PATH'], 'w')
arr = []
for _ in range(6):
arr.append(list(map(int, input().rstrip().split())))
result = hourglassSum(arr)
fptr.write(str(result) + '\n')
fptr.close()
Editor is loading...
Leave a Comment