[SWEA] 2001 파리퇴치 (파이썬/python)

🎈문제

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=2&problemLevel=3&problemLevel=4&contestProbId=AV5PzOCKAigDFAUq&categoryId=AV5PzOCKAigDFAUq&categoryType=CODE&problemTitle=&orderBy=INQUERY_COUNT&selectCodeLang=JAVA&select-1=4&pageSize=10&pageIndex=1

 

🎁알고리즘 및 접근

구현

이런식으로 행렬을 훑으면서 지나가는 방법은 딥러닝에서 convolution을 공부할 때 많이 봤던 것이여서 반가웠다.

 

💻코드

# sw 2001 파리퇴치

T = int(input())
for test_case in range(1, T + 1):
    n,m = map(int,input().split())
    lst = [list(map(int,input().split())) for _ in range(n)]
    answer = 0

    for i in range(n-m+1):
        for j in range(n-m+1):
            cnt = 0
            for a in range(i,m+i):
                for b in range(j,m+j):
                    cnt += lst[a][b]
            answer = max(answer,cnt)
     
    print(f"#{test_case} {answer}")

🎄코드 설명

간단한 문제여서 설명은 생략한다.