[Python][SWEA] 959. 두 개의 숫자열 D2

 

SW Expert Academy

 

---

🗒️파이썬 코드 풀이

T = int(input())
for tc in range(1,T+1):
    N,M = map(int,input().split()) # M,N 각각 입력
    N_lst = list(map(int,input().split())) # N 리스트 입력
    M_lst = list(map(int,input().split())) # M 리스트 입력

    result = []
    diff_len = abs(N-M)  # M과 N 배열의 길이 차이

    if N >= M:   # N과 M 중 작은 배열의 크기를  큰 배열 크기만큼 0로 채워주기
        lst = N_lst  # N과 M 중에 더 큰 배열이 lst가 됨
        for i in range(0,diff_len+1): #  모든 경우의 수 채워주기
            one_padding = [0] * N   # value가 0인 배열 생성 (큰 배열 기준 크기 생성)
            one_padding[i:i+M] = M_lst
            result.append(one_padding)
    else : 
        lst = M_lst
        for i in range(0,diff_len+1):
            one_padding = [0] * M
            one_padding[i:i+N] = N_lst
            result.append(one_padding)
    
    ans = []
    for ls in result :  # 큰 배열과 작은 배열(0로 채워진) 곱하고 저장
        sum = 0  
        for i in range(len(lst)) :
            sum += (lst[i] * ls[i])
        ans.append(sum)
    print(f"#{tc} {sorted(ans)[-1]}")

1. 작은 배열을 큰 배열 크기만큼 0으로 채워주기

2. 0으로 채워준 작은 배열들을 reuslt 리스트에 담아주기

3. result 만큼 반복하여  (0으로 채워준 배열 * 큰 배열) 

4. 결과를 저장하고 sort하여 출력

 

📌개선점 

   if N<M :
        N_lst,M_lst = M_lst,N_lst

N이 항상 크다고 가정하고, 아닌 경우 swap을 해줬으면 코드가 더 간단해졌을 것 같다.

 

그리고 굳이 작은 배열에 0으로 패딩하지 말고 필요한 것만 계산하면 됐다.

 

---

 

📚문제

 

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부터 시작한다.)