[Algorithm] Hackerrank - 2D Array - DS

텍스트를 복사해서 넣으니 수식이 복사가 안되는군ㅠ

문제:

코드:

#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the hourglassSum function below.
def hourglassSum(arr):
    sum_ = -9999
    for y in range(len(arr)-2):
        for x in range(len(arr[0])-2):
            temp = arr[y][x] + arr[y][x+1] + arr[y][x+2] +\
                            + arr[y+1][x+1] +\
                arr[y+2][x] + arr[y+2][x+1] + arr[y+2][x+2]
            if temp > sum_:
                sum_ = temp
    return 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()

 문제점:

만약에 배열이 길면 해당코드는 사용하지못함.. 많은시간이 소요될것으로 예상됨