[python/inflearn/section3] 사과나무(다이아몬드)

[python/inflearn/section3] 사과나무(다이아몬드)

본 내용은 파이썬 알고리즘 문제풀이 (코딩테스트 대비) 강의를 토대로 작성하였습니다.

이미지 클릭시, 강의로 이동됩니닷 :)

# 사과나무(다이아몬드)

import sys
#sys.stdin=open("input.txt", "r")
n = int(input())
a = [list(map(int, input().split())) for _ in range(n)]
# 가운데를 기준으로 시작값과 끝 값을 구한다.
mid = n//2
s = e = mid
res = 0
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)