현수의 농장은 N*N 격자판으로 이루어져 있으며, 각 격자안에는 한 그루의 사과나무가 심어저 있다. N의 크기는 항상 홀수이다. 가을이 되어 사과를 수확해야 하는데 현수는 격자판안의 사 과를 수확할 때 다이아몬드 모양의 격자판만 수확하고 나머지 격자안의 사과는 새들을 위해서 남겨놓는다. 만약 N이 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 |
현수과 수확하는 사과의 총 개수를 출력하세요.
▣ 입력설명
첫 줄에 자연수 N(홀수)이 주어진다.(3<=N<=20) 두 번째 줄부터 N줄에 걸쳐 각 줄에 N개의 자연수가 주어진다. 이 자연수는 각 격자안에 있는 사과나무에 열린 사과의 개수이다. 각 격자안의 사과의 개수는 100을 넘지 않는다.
▣ 출력설명
수확한 사과의 총 개수를 출력합니다.
▣ 입력예제
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
▣ 출력예제
379
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)
이전에 했던 격자판과 형식은 동일합니다.
res라는 변수에 계속 누적하도록 0으로 초기화 해줍니다.
s와 e라는 변수를 n개 행의 절반(s=e=n//2)만큼 이라고 해준다면,
위와 같은 경우는 5개의 행의 절반이기 때문에 2번 index에서 반복문을 돌게됩니다.
그 상태에서 s가 1 감소하고, e가 1 증가한다면 s는 1번index, e는 3번index까지 펼쳐지는 것을 반복합니다.
탐색을 다 한 뒤, if문에서 행의 절반보다 작아진다면, 다시 줄어들도록 설정해줍니다.
379