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

[문제 링크]

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

SW Expert Academy

---

[문제 설명]

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 i in range(1, t+1):
	# 입력받기
    n, m = map(int, input().split())
    a = list(map(int, input().split()))
    b = list(map(int, input().split()))
    res = []
    
    # 더 작은 리스트를 aj, 큰 리스트를 bj에 저장
    if n > m:
        aj = b
        bj = a
        n, m = m, n
    else:
        aj = a
        bj = b
    
    # 큰 리스트의 시작 인덱스를 하나씩 옮겨가며 계산값 저장
    for j in range(m-n+1):
        cal = sum([aj[q]*bj[j+q] for q in range(n)])
        res.append(cal)
     
    print('#'+str(i), max(res))

 

풀이

• 작은 리스트에 들어있는 수는 모두 곱해지고, 큰 리스트도 작은 리스트 길이만큼 수를 곱하게 되므로 큰 리스트의 시작 인덱스를 하나씩 증가시켜 옮겨가며 계산한다.

---

알아둘 사항

•