SW Expert Academy

2001번 파리 퇴치 / python

문제

풀이과정

파리채의 크기 만큼 완전 탐색
파리채의 크기 만큼 탐색시 탐색의 범위 지정

소스코드

T = int(input())

result = []
for i in range(T):
    n, m = map(int, input().split())
    fly = []
    for _ in range(n):
        arr = list(map(int, input().split()))
        fly.append(arr)

    kill = 0
    for x in range(n - m + 1): # 파리채의 영역 접근, 인덱스 계산에 주의
        for y in range(n - m + 1):
            temp = 0
            for a in range(m):
                for b in range(m):
                    temp += fly[x + a][y + b]
            if temp > kill:
                kill = temp

    result.append(kill)

for i in range(T):
    print("#{} {}".format((i + 1), result[i]))