[SWEA][Python] 1959 두 개의 숫자열

[문제]

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

SW Expert Academy

 

 

[코드]

abs 사용해서 M-N 하니 인덱스 초과가 발생해서 

if N > M 으로 따로 빼주었다. 

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

    if N > M:
        A, B = B, A
        N, M = M, N

    ans = []
    for i in range(0,M-N+1):
        tmp = 0
        for j in range(N):
            tmp += A[j] * B[i+j]
        ans.append(tmp)

    print('#'+str(tc),max(ans))