SWEA: 2805 농작물 수확하기

문제

SW Expert Academy

 

문제 풀이

 농작물을 수확하여 얻을 수 있는 이익이 행열에 주어 질 때, 마름모로 탐색하여 이익의 합을 구하는 문제이다. 마름모로 접근하는 로직만 설계하면 끝나는 문제이다. 마름모를 접근하기 위해서는 다음과 같이 생각하면 된다.

 

 예를 들어 3x3 행렬이라고 하면 다음과 같이 접근하게 된다.

• (0, 1)
• (1, 0), (1, 1), (1, 2)
• (2, 1)

 이와 같이 접근하고자 할 때 `j`의 범위에 대한 규칙은 `abs(N // 2 - i)`부터 `abs(N - half)`이다. `half`는 i가 절반이 되기 전까지는 감소하고, 절반이 된 이후로는 증가하여 마름모와 같은 접근이 가능하다.

 

코드

T = int(input())
for test_case in range(1, T + 1):
    answer = 0
    n = int(input())
    farm = [list(map(int, input().strip())) for _ in range(n)]
    half = n // 2

    back = half

    for i in range(n):
        for j in range(abs(half - i), abs(n - back)):
            answer += farm[i][j]

        back = back - 1 if i < half else back + 1
    print('#' + str(test_case), answer)