[SWEA] 2001. 파리 퇴치 [D2]

📚 문제

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV5PzOCKAigDFAUq&categoryId=AV5PzOCKAigDFAUq&categoryType=CODE&problemTitle=2001&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=&pageSize=10&pageIndex=1

이중 for문으로 합을 전부 확인하며 가장 큰 합을 출력한다.

📒 코드

T = int(input())
for tc in range(1, 1+T):
    N, M = map(int, input().split())
    flies = [list(map(int, input().split())) for _ in range(N)] # 파리들이 들어있는 리스트
    max_sum = 0
    for i in range(N-M+1):  # 탐색할 MxM의 맨 위 인덱스를 택한다.
        for j in range(N-M+1):
            sum_fly = 0
            for k in range(M):  # 고른 인덱스를 중심으로 MxM의 네모 영역의 합을 구한다.
                for l in range(M):
                    sum_fly += flies[i+k][j+l]
            if max_sum < sum_fly:
                max_sum = sum_fly
    print(f'#{tc} {max_sum}')

🔍 결과 : Pass