[탐색&시물레이션] - 사과나무(다이아몬드)

사과나무(다이아몬드)

현수의 농장은 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

 

나의 코드

n=int(input())
arr=[list(map(int,input().split())) for _ in range(n)]
S=0

for i in range(n//2+1):
    S+=sum(arr[i][n//2-i:n//2+1+i])
for i in range(n//2+1,n):
    S+=sum(arr[i][n//2+i-(n-1):n//2-i+(n)])

print(S)

 

음.. 이거 너무 규칙성만 무지성으로 찾아서 풀긴했는데...

다른 가독성이 좋은 코드를 참조해야겠따..

이번 코드는 코드 수는 짧지만 이해적으로는 좋지 않은것 같다..!

 

정답 코드

import sys
sys.stdin = open("input.txt", 'r')
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)

 

오.. 코드량은 나의 코드 보다 길지만 훠얼씬 이해하기 쉬운 코드 같다..!

좋은 코드 같으니 기억해야쥥