[python] 코딩테스트 대비 - 2차원배열 다이아몬드

문제)

답안 코드)

import sys
#sys.stdin = open("input.txt", 'r')
n=int(input())
a=[list(map(int, input().split())) for _ in range(n)]
res=0#정답
s=e=n//2#초기값(격자판 가운데 최상단)
 
for i in range(n):
    for j in range(s, e+1):
        res+=a[i][j]
        
    if i<n//2:#중간까지 범위 넓히기
        s-=1
        e+=1
    else:#중간 지나면 범위 쫍히기
        s+=1
        e-=1
print(res)