[SWEA Python] 2001. 파리 퇴치

문제링크

 

풀이

T = int(input())
for test in range(1, T+1):
    n, m = map(int, input().split())
    arr = [0 for _ in range(n)]
    arr = [list(map(int, input().split())) for _ in range(n)]
    res_list = []
    for i in range(n-m+1):
        for j in range(n-m+1):
            res = 0
            for x in range(m):
                for y in range(m):
                    res = res + arr[x+i][y+j]
            res_list.append(res)
    print(f"#{test} {max(res_list)}")

2차원 배열 입력받기 arr = [list(map(int, input().split())) for _ in range(n)]

지금 다시 코드를 보면 4번째 줄을 굳이 써야 하나 싶다.

파리채는 배열의 범위를 넘지 않는 n-m+1만큼 수행