[파이썬 알고리즘] 사과나무(다이아몬드)

사과나무(다이아몬드)

📄 문제

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

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

▣ 입력설명
첫 줄에 자연수 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 (홀수), 두 번째 줄부터 N번째 줄까지 N개의 자연수
# 출력: 수확한 총 사과 개수

'''
만약 7이면,
1 3 5 7 5 3 1

만약 9이면,
1 3 5 7 9 7 5 3 1
'''

n = int(input())                                                            # 정수 n 입력 및 할당
array = []                                                                  # 빈 array 할당
list_s = []                                                                 # list_s 할당

for i in range(n):                                                          # n만큼 반복
    array.append(list(map(int,input().split())))                            # array에 각 행의 숫자 추가

for i in range(n):                                                          # n만큼 반복
    if i <= int(n/2):                                                       # 위의 절반 줄의 경우에는
        list_s.append(sum(array[i][int(n/2)-i:int(n/2)+1+i]))               # 해당 조건만큼만 슬라이싱해서 더함
    else:                                                                   # 반대로 밑의 절반 줄의 경우에는
        list_s.append(sum(array[i][int(n/2)-(n-i-1):int(n/2)+1+(n-i-1)]))   # 해당 조건만큼만 슬라이싱해서 더함

print(sum(list_s))                                                          # 합을 구하여 출력

 

다른 사람 풀이

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)