2805. 농작물 수확하기

문제

농장의 크기 N와 농작물의 가치가 주어질 때, 규칙에 따라 얻을 수 있는 수익은 얼마인지 구하여라.

코드

# 0828 5/50 error
import sys
sys.stdin = open('input.txt')
# 농장의 크기 N와 농작물의 가치가 주어질 때,
# 규칙에 따라 얻을 수 있는 수익은 얼마인지 구하여라.

T = int(input())
for tc in range(1,T+1):
    # N x N 농장 (항상 홀수)
    N = int(input())
    arr = [list(map(int, input())) for _ in range(N)]
    center = N//2
    result = []
    ans = 0

   # 가운데줄(+)은 전부 넣기
    for a in range(N):
        result.append(arr[center][a])
        result.append(arr[a][center])

    # 가운데 중복부분 제거
    ans -= arr[center][center]
    
    # 가운데 기준 위쪽/아래쪽 부분
    for i in range(center):
        if i != 0:
            for k in range(center-i, center+i+1):
                if k != center:
                    result.append(arr[i][k])
                    result.append(arr[N-1-i][k])

    for j in range(len(result)):
        ans += result[j]

    print("#{} {}".format(tc, ans))

장애물

1. 어떻게 풀지 아이디어가 빨리 오지 않았다.

2. 센터기준 홀수개씩 입력시키는 코드 짜는 부분에서 가장 시간을 오래보냈다.