[예제] [파이썬] [탐색] 사과나무(다이아몬드)

현수의 농장은 N*N 격자판으로 이루어져 있으며, 각 격자안에는 한 그루의 사과나무가 심어저
있다. N의 크기는 항상 홀수이다. 가을이 되어 사과를 수확해야 하는데 현수는 격자판안의 사
과를 수확할 때 다이아몬드 모양의 격자판만 수확하고 나머지 격자안의 사과는 새들을 위해서
남겨놓는다.
만약 N이 5이면 아래 그림과 같이 진한 부분의 사과를 수확한다.

 

 

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

 

▣ 입력설명
첫 줄에 자연수 N(홀수)이 주어진다.(3<=N<=20)
두 번째 줄부터 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
379

 

---

 

import sys
sys.stdin=open('input.txt','r')

if __name__=="__main__":
    N=int(input())
    board=[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+=board[i][j]
        
        if i<N//2:      # 행이 절반이상 안갔을때
            s-=1        # 열 시작점을 왼쪽으로 한칸씩
            e+=1        # 열 끝점을 오른쪽으로 한칸씩 
        else:           # 행이 절반 이상 갔을때
            s+=1        # 반대로 해주면 됨
            e-=1
    
    print(res)