본문 바로가기
알고리즘/문제풀이

sw Expert Academy D2 파리퇴치

by 새싹감자 2022. 7. 29.

이중 배열과 for문을 사용해서 해당 위치만큼 값을 더한 후에 max값을 바꿔주는 형식으로 코드를 작성하였다.

for문을 좀더 적게 쓸 수 있도록 코드를 다시 짜봐야겠다.

num=int(input())
for z in range(num):
        n,m=map(int,input().split())
        bug=[[0*n]]*n
        max=0
        result=0
        resultin=[]
        for i in range(n):
         bug[i]=list(map(int,input().split()))


        for i in range(n-m+1):
            for j in range(n-m+1):
                for a in range(i,i+m):
                    for b in range(j,j+m):
                        result = result+bug[a][b]

                if (max < result):
                  max=result
                result=0
        print('#{}'.format(z+1),end=' ')
        print(max)

댓글