[SWEA D2] 1959 두 개의 숫자열 (Python)

※ SW Expert 아카데미의 문제를 무단 복제하는 것을 금지합니다.

문제

SW Expert Academy

 

풀이

• N 개의 숫자로 구성된 숫자열 Ai (i=1~N)
• M 개의 숫자로 구성된 숫자열 Bj (j=1~M)
• Ai 나 Bj  를 자유롭게 움직여서 숫자들이 서로 마주보는 위치를 변경할 수 있지만 더 긴 쪽의 양끝을 벗어나서는 안 된다.
• 서로 마주보는 숫자들을 곱한 뒤 모두 더할 때 최댓값을 구한다.
• 숫자열의 길이) 비교해서 더 긴 것 짧은 것을 구분해준다.
• 결과값을 저장하는 total 변수에는 - inf 로 초기화

T = int(input())
# 여러개의 테스트 케이스가 주어지므로, 각각을 처리합니다.
for test_case in range(1, T + 1):
    n, m = map(int, input().split())
    total =-float('inf')
    if n <= m :
        short, long = n, m
        A = list(map(int, input().split()))
        B = list(map(int, input().split()))
    else:
        short, long = m, n
        B = list(map(int, input().split()))
        A = list(map(int, input().split()))    
    for i in range(long - short + 1):
        temp = 0
        for j in range(short):
            temp += A[j] * B[i+j]
        if total <= temp:
            total = temp

    print(f"#{test_case} {total}")