2805. 농작물 수확하기

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV7GLXqKAWYDFAXB 

SW Expert Academy

T = int(input())
for tc in range(1,T+1):
    n = int(input())
    arr = [list(map(int,input())) for _ in range(n)]
    revenue = 0
    k = 2
    for j in range(n):
        if 0<= n//2-j and n//2+j < n:  # 전환점
            for i in range(n//2-j,n//2+j+1):
                revenue += arr[j][i]
        else:
            for i in range(n//2-j+k,n//2+j-k+1):
                revenue += arr[j][i]
            k += 2  # 양쪽 2씩 점점 줄어야되므로
    print(f'#{tc} {revenue}')