[Python/알고리즘] 격자판 최대합

문제

5*5 격자판에 아래롸 같이 숫자가 적혀있습니다.

10	13	10	12	15
12	39	30	23	11
11	25	50	53	15
19	27	29	37	27
19	13	30	13	19

N*N의 격자판이 주어지면 각 행의 합, 각 열의 합, 두 대각선의 합 중 가장 큰 합을 출력합 니다. 

 

입력

첫 줄에 자연수 N이 주어진다.(1<=N<=50) 두 번째 줄부터 N줄에 걸쳐 각 줄에 N개의 자연수가 주어진다. 각 자연수는 100을 넘지 않는다.

5 10 13 10 12 15 12 39 30 23 11 11 25 50 53 15 19 27 29 37 27 19 13 30 13 19

 

출력

최대합을 출력합니다.

155

 

답

N = int(input())

li = []
for i in range(N):
    li.append(list(map(int, input().split())))

m = 0
s_lr = 0  # 대각선 위에서 아래
s_rl = 0  # 대각선 아래에서 위

for i in range(N):
    s_r = 0  # 가로
    s_c = 0  # 세로

    for j in range(N):
        s_r += li[i][j]
        s_c += li[j][i]

    if s_r > m:
        m = s_r
    if s_c > m:
        m = s_c
    
    s_lr += li[i][i]
    s_rl += li[N - i - 1][i]

if s_lr > m:
    m = s_lr
if s_rl > m:
    m = s_rl
    
print(m)