[swea] 2001. 파리퇴치 : python

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV5PzOCKAigDFAUq 

SW Expert Academy

---

T = int(input())

for tc in range(1, T+1):
    N, M = map(int, input().split())

    fly = [list(map(int,input().split())) for _ in range(N)]
    attack = [[0]*M for __ in range(M)]
    max_num = 0

    for i in range(N-M+1):
        for j in range(N-M+1):
            add_num = 0
            for fly_i in range(M):
                for fly_j in range(M):
                    add_num += fly[i+fly_i][j+fly_j]
                    if add_num > max_num:
                        # print(add_num)
                        max_num = add_num

    print('#{} {}'.format(tc, max_num))

2차원 배열을 입력받고 접근 할 수 있어야 풀 수 있는 문제!

모기가 있는 2차원 배열을 만들고

파리채 2차원 배열을 만들어 파리채의 (0, 0)지점부터 파리채 크기만큼의 구역에 모기가 몇마리 있는지 계산 후

max_num에 넣고 출력한다