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

 

 

코드 : 

# 격자판 최대합

import sys

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

N = int(sys.stdin.readline())

arr = [list(map(int, sys.stdin.readline().split())) for _ in range(N)] # 5개 행 

max = -214700000

for i in range(N):
    sum1 = sum2 = 0
    for j in range(N):
        sum1 += arr[i][j]
        sum2 += arr[j][i]
    if sum1 > max:
        max = sum1
    if sum2 > max:
        max = sum2

sum1 = sum2 = 0

for i in range(N):
    sum1 += arr[i][i]
    sum2 += arr[i][N-i-1] # index 

if sum1 > max:
    max = sum1 
if sum2 > max:
    max = sum2

print(max)