[SWEA - D3] 2805. 농작물 수확하기

 

 

 

 

나의 코드

t = int(input())
for tc in range(1,t+1):
    n = int(input())
    lt = n//2
    rt = n//2
    result = 0
    for _ in range(n//2):
        arr = list(map(int,input()))
        result += sum(arr[lt:rt+1])
        lt -= 1
        rt += 1
    for _ in range(n//2,n):
        arr = list(map(int,input()))
        result += sum(arr[lt:rt+1])
        lt += 1
        rt -= 1
    print("#"+str(tc) , result )

 

코드가 너무 지저분해서 다시 풀었다.

 

수정코드 

t = int(input())
for tc in range(1,t+1):
    n = int(input())
    s=e=n//2
    res = 0
    for i in range(n):
        arr = list(map(int,input()))
        res += sum(arr[s:e+1])
        if i < n//2 :
            s -= 1
            e += 1
        else:
            s += 1
            e -= 1
    print("#"+str(tc) , res )

 

시작 지점 s 와 끝지점 e 를 만듦

농장의 중앙지점 n//2 로 오면 반대로 크기 조정 

 

 

/ 와 // 는 다르다 .