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

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=2&contestProbId=AV5PpoFaAS4DFAUq&categoryId=AV5PpoFaAS4DFAUq&categoryType=CODE&problemTitle=&orderBy=INQUERY_COUNT&selectCodeLang=PYTHON&select-1=2&pageSize=10&pageIndex=2 

 

SW Expert Academy

 

j의 구간을 잘 못 설정하여 계속 인덱스 오류가 떴는데 마침내 해결하였다.

 

t = int(input())

for i in range(1, t+1):
    a , b = map(int, input().split())
    a_lst = list(map(int, input().split()))
    b_lst = list(map(int, input().split()))
    mi = min(a, b)
    ma = max(a, b)
    
    if a > b:
        a_lst, b_lst = b_lst, a_lst

    result = []
    
    for j in range(ma - mi + 1):
        current_sum = 0
        for k in range(mi):
            current_sum += a_lst[k] * b_lst[j+k]
        result.append(current_sum)

    print("#{} {}".format(i, max(result)))

 

혼자 공부하며 올리는 블로그입니다. 틀린 내용은 댓글 남겨주시면 감사하겠습니다.