[Python] SWEA D2 2001번 - 파리 퇴치

SW Expert Academy

T = int(input())
for t in range(1, T + 1):
    n, m = map(int, input().split())
    space = [list(map(int, input().split())) for _ in range(n)]
    ans = 0
    for i in range(n - m + 1):
        for j in range(n - m + 1):
            tmp = 0
            for k in range(m):
                for r in range(m):
                    tmp += space[i + k][j + r]
            ans = max(ans, tmp)
    print('#%d %d' % (t, ans))

- 앞선 문제인 스도쿠 검증의 아이디어와 똑같은 문제입니다.

- n x n 사이즈의 공간에 m x m 사이즈에 해당하는 인덱스의 합을 구해야 합니다.

- for문의 범위를 벗어나지 않기 위해 n - m + 1로 전체 탐색 범위를 제한해 줍니다. 그 이유는 m의 사이즈에 따라 탐색하는 범위가 결정되기 때문입니다. 

 

피드백은 언제나 환영합니다.