[SWEA #1959] 두 개의 숫자열

문제

N 개의 숫자로 구성된 숫자열 Ai (i=1~N) 와 M 개의 숫자로 구성된 숫자열 Bj (j=1~M) 가 있다.
아래는 N =3 인 Ai 와 M = 5 인 Bj 의 예이다.

Ai 나 Bj 를 자유롭게 움직여서 숫자들이 서로 마주보는 위치를 변경할 수 있다.
단, 더 긴 쪽의 양끝을 벗어나서는 안 된다.

서로 마주보는 숫자들을 곱한 뒤 모두 더할 때 최댓값을 구하라.
위 예제의 정답은 아래와 같이 30 이 된다.

제약 사항
N 과 M은 3 이상 20 이하이다.

입력
가장 첫 줄에는 테스트 케이스의 개수 T가 주어지고, 그 아래로 각 테스트 케이스가 주어진다.
각 테스트 케이스의 첫 번째 줄에 N 과 M 이 주어지고,
두 번째 줄에는 Ai,
세 번째 줄에는 Bj 가 주어진다.

출력
출력의 각 줄은 '#t'로 시작하고, 공백을 한 칸 둔 다음 정답을 출력한다.
(t는 테스트 케이스의 번호를 의미하며 1부터 시작한다.)

 

내가 쓴 코드

T = int(input())
for t in range(T):
    N, M = map(int, input().split())
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))
    if M < N:
        N, M = M, N
        A, B = B, A
    # N < M
    max_sum = -999999
    for i in range(M - N + 1):
        sum = 0
        for n in range(N):
            sum += A[n] * B[i + n]
        if sum > max_sum:
            max_sum = sum

    print(f'#{t+1} {max_sum}')

아니 다 했는데, 마지막에 저 if문을 쓰니까 답이 안나오는거다... 그래서 한참 헤메고, 강사님 코드도 봤는데 진짜 똑디인거다.. 후하! 그래서 결국 강사님한테 여쭤봤는데, 등호가 저게 맞냐고 그러는거다...ㅠㅠ 힝.... 등호를 잘못 써서 안돌아가고있었다..^^^^ 하하 이런 바보가 또 없지 ^^*

 

출력 결과

#1 30
#2 63
#3 140
#4 181
#5 63
#6 58
#7 22
#8 120
#9 96
#10 70