[SWEA / python] 1959. 두 개의 숫자열

 

import sys
sys.stdin = open("input.txt", "r")

T = int(input())

for test_case in range(1, T + 1):
    n , m = map(int,input().split()) #A, B배열 크기

    a = list(map(int,input().split()))
    b = list(map(int, input().split()))

    result = 0 #최댓값

    if(m>n):
        for i in range(m-n+1):
            sum = 0
            for k in range(n):
                sum += a[k]*b[i+k]

            if (sum > result):
                result = sum

    else:
        for i in range(n-m+1):
            sum = 0
            for k in range(m):
                sum += b[k]*a[i+k]

            if (sum > result):
                result = sum

    print("#"+str(test_case)+" "+str(result))

sum 초기화하는 부분을 잘못해서 헤메고있었네,,, 바보