본문 바로가기
problem solving

2805. 농작물 수확하기

by ormgm 2019. 9. 29.

import copy
t = int(input())
for q in range(t):
    n = int(input())
    arr = []
    for i in range(n):
        arr.append(list(input()))
    half = int(n/2)
    row = copy.deepcopy(half)
    row_end = copy.deepcopy(half)
    sw = 0
    ans = 0
    if n == 1:
        print('#%d'%(q + 1), arr[0][0])
        continue
    for i in range(n):
        for j in range(row,row_end+1):
            ans = ans + int(arr[i][j])
        if sw == 0 :
            row -= 1
            row_end += 1
            if row_end == n - 1:
                sw = 1
        elif sw == 1:
            row += 1
            row_end -= 1
    print('#%d'%(q+1),ans)

댓글