Algorithm/BOJ_14500.py at main · DebugMind/Algorithm · GitHub

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
39
40
41
42
43
44
45
46
47
48
49
R,C = map(int, input().split())
board = []
for _ in range(R) :
    board.append(list(map(int, input().split())))

ans = 0

for r in range(R) :
    for c in range(C-3) :
        ans = max(ans, sum(board[r][c:c+4]))

for r in range(R-3) :
    for c in range(C) :
        tmp = 0
        for i in range(4) : tmp += board[r+i][c]
        ans = max(ans, tmp)

for r in range(R-1) :
    for c in range(C-1) :
        tmp = 0
        for i in range(2) :
            for j in range(2) :
                tmp += board[r+i][c+j]
        ans = max(ans, tmp)

arr1 = [[[0,0],[1,0]], [[0,1],[1,1]], [[1,0],[2,0]], [[1,1],[2,1]], [[0,0],[2,0]], [[0,1],[2,1]], [[0,0],[2,1]], [[0,1],[2,0]]]
for r in range(R-2) :
    for c in range(C-1) :
        tmp = 0
        for i in range(3) :
            for j in range(2) :
                tmp += board[r+i][c+j]

        for one, two in arr1 :
            ans = max(ans, tmp - board[r+one[0]][c+one[1]] - board[r+two[0]][c+two[1]])


arr2 = [[[0,0],[0,1]], [[0,1],[0,2]], [[1,0],[1,1]], [[1,1],[1,2]],[[0,0],[0,2]], [[1,0],[1,2]], [[0,0],[1,2]], [[1,0],[0,2]]]
for r in range(R-1) :
    for c in range(C-2) :
        tmp = 0
        for i in range(2) :
            for j in range(3) :
                tmp += board[r+i][c+j]

        for one, two in arr2 :
            ans = max(ans, tmp - board[r+one[0]][c+one[1]] - board[r+two[0]][c+two[1]])

print(ans)