[알각코] 인프런 3-7 사과나무

사과나무 (다이아몬드)

문제 정보는 인프런의 파이썬 알고리즘 문제풀이 코딩테스트에 있습니다!

문제풀이

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)

2차원배열 기본 리스트에서 행이나 열만 for문 도는 횟수를 늘리고 줄임으로써 효율적으로 이중포문을 돌도록 풀이