[Python] 격자판 최대합

[문제]

 

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

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

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

▣ 출력설명
최대합을 출력합니다.

▣ 입력예제 1
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

▣ 출력예제 1
155

---

[풀이]

n = int(input())
a = [list(map(int, input().split())) for _ in range(n)]

# 가장 작은 값으로 최댓값 할당
largest = -2147000000

for i in range(n):
    sum1 = sum2 = 0
    for j in range(n):
        sum1 += a[i][j] # 행의 합
        sum2 += a[j][i] # 열의 합

    if sum1 > largest:
        largest = sum1
    if sum2 > largest:
        largest = sum2

sum1 = sum2 = 0
for i in range(n):
    sum1 += a[i][i]
    sum2 += a[i][n-i-1]

    if sum1 > largest:
        largest = sum1
    if sum2 > largest:
        largest = sum2

print(largest)

---

[내용]

1. 리스트를 반복해서 input 할 경우에는 a = [list(map(int, input().split))) for _ in range(n)] 을 사용하면 한 줄로 가능하다

2. 처음 리스트 받을 때  2차원 배열로 받은게 아니라서 a[i][j] 이런식의 표현식을 사용할 수가 없었다.

3. 행렬 표현식 => a[열][행]