문제 / 사과나무(다이아몬드) / Python 파이썬

 

문제 제목: 사과나무(다이아몬드)

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

---

나의 풀이

n = int(input())
arr = [list(map(int, input().split())) for _ in range(n)] # 2차원 배열 입력
sum_arr = [] # 답을 넣어줄 배열
mid = n // 2 # 중간 값
cnt = 0
pt = 1
answer = 0
while cnt != n:

    if cnt <= mid:
        sum_arr.append(arr[cnt][mid-cnt:mid+1+cnt])

    else:
        sum_arr.append(arr[cnt][pt:n-pt])
        pt += 1

    cnt += 1

for i in sum_arr: # 2차원 배열의 합을 구한다
    tmp = sum(i)
    answer += tmp
print(answer)

모범답안

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)

 

 

---

 

오랜만에 문제를 푸는 거라서 너무 복잡하게 생각한거 같다....................

s와 e라는 변수를 두개 두어서 하나는 증가, 하나는 감소 로 함으로 문제를 풀면 된다.