[Algorithm] 백준 파이썬 14500 테트로미노

n, m = map(int, input().split())
data = []
for i in range(n):
    data.append(list(map(int, input().split())))

shapeList = [[[0, 1], [0, 2], [0, 3]],  # ㅡ
             [[1, 0], [2, 0], [3, 0]],  # ㅣ
             [[0, 1], [1, 0], [1, 1]],
             [[1, 0], [2, 0], [2, 1]],
             [[1, 0], [2, 0], [2, -1]],
             [[0, 1], [0, 2], [1, 0]],
             [[0, 1], [0, 2], [1, 2]],
             [[0, 1], [1, 1], [2, 1]],
             [[0, 1], [1, 0], [2, 0]],
             [[0, 1], [0, 2], [-1, 2]],
             [[1, 0], [1, 1], [1, 2]],
             [[1, 0], [1, 1], [2, 1]],
             [[1, 0], [1, -1], [2, -1]],
             [[0, 1], [-1, 1], [-1, 2]],
             [[0, 1], [1, 1], [1, 2]],
             [[0, 1], [0, 2], [1, 1]],
             [[1, 0], [1, 1], [2, 0]],
             [[1, 0], [1, -1], [2, 0]],
             [[0, 1], [0, 2], [-1, 1]]]

max_sumary = 0
sumary = 0
for i in range(n):
    for j in range(m):
        for shape in shapeList:
            sumary = data[i][j]
            for k in range(3):
                nX = j + shape[k][1]
                nY = i + shape[k][0]
                if 0 <= nX <= m - 1 and 0 <= nY <= n - 1:
                    sumary += data[nY][nX]
            max_sumary = max(max_sumary, sumary)

print(max_sumary)

처음에는 모양에(ㅣㅡㅏㄹ..) 따라 나눠서 구하려고 했는데 왕실의 나이트 문제가 생각나 19가지 경우를 list안에 넣어 구해보았다.