사과나무(다이아몬드)
현수의 농장은 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
코드
인덱스 만으로
3
2 4
1 5
의 숫자를 꺼내려고 하니 어려움이 있었다.
인덱스를 나타내는 수를 하나의 변수(a, b)로 설정하고
그 값에 변화를 주면서 값을 뽑아 내는 것이 중요하다.
계속 변하는 숫자를 for문 안의 인덱스만으로 표현하는 것은 매우 어렵기 때문에 간단하지만 중요한 스킬인 것 같다.
인덱스를 나태내는 수가 감소하는 지점이 언제까지 인지를 아는 것도 조금만 생각해보면 알 수 있다.
N = int(input())
arr = [list(map(int, input().split())) for _ in range(N)]
a = N//2 # 인덱스를 나타내는 숫자를 변수로 저장하기
b = N//2 + 1 # 변하는 숫자를 for문 안의 인덱스만으로 표현하는 것은 매우 어렵기 때문에 중요한 스킬이다.
res = 0
for i in range(N):
res += sum(arr[i][a:b])
if i < N//2: # 중간 까지만 감소한다. 그 뒤로 다시 올라감
a-=1
b+=1
else:
a+=1
b-=1
print(res)
'''
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
'''