본문 바로가기

파이썬 알고리즘 풀이

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

문제

현수가 수확하는 사과의 총 개수를 출력하세요.

 

조건

현수의 농장은 N*N 격자판으로 이루어져 있으며, 각 격자안에는 한 그루의 사과나무가 심어저 있다.

 

N의 크기는 항상 홀수이다

 

가을이 되어 사과를 수확해야 하는데 현수는 격자판안의 사 과를 수확할 때 다이아몬드 모양의 격자판만 수확하고 나머지 격자안의 사과는 새들을 위해서 남겨놓는다.

 

만약 N이 5이면 아래 그림과 같이 진한 부분의 사과를 수확한다.

제한사항

첫 줄에 자연수 N(홀수)이 주어진다.(3<=N<=20) 두 번째 줄부터 N줄에 걸쳐 각 줄에 N개의 자연수가 주어진다.

 

이 자연수는 각 격자안에 있는 사과나무에 열린 사과의 개수이다.

 

각 격자안의 사과의 개수는 100을 넘지 않는다.

 

수확한 사과의 총 개수를 출력합니다.

 

문제풀이

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

s=e=N//2
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)

'파이썬 알고리즘 풀이' 카테고리의 다른 글

19) 봉우리  (0) 2021.10.22
18) 곳감(모래시계)  (0) 2021.10.22
16) 격자판 최대합  (0) 2021.10.22
15) 수들의 합  (0) 2021.10.20
14) 두 리스트 합치기  (0) 2021.10.20