[SW Academy] 1959. 두 개의 숫자열

DreamJJW·2023년 5월 30일
0

SW Academy

목록 보기
13/26

📖 문제


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

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



📖 풀이


문제처럼 배열의 인덱스를 움직여가며 푸려면 난해할 듯하여, 입력 받는 두 리스트 중에 더 긴 쪽의 0번 인덱스를 pop해가며 합을 구하는 방식으로 접근하였다.

이 문제에서 조심해야할 점은, 배열 N이 배열 M보다 짧을 수도, 길 수도 있다는 점인데(역 또한 성립), 이 점을 고려하지 않는다면 인덱스 에러에 봉착할 수 있다.

그 점만 조심한다면 어렵지 않은 문제인듯 하다.


T = int(input())

# 여러개의 테스트 케이스가 주어지므로, 각각을 처리합니다.
for test_case in range(1, T + 1):
    N, M = map(int, input().split())
    shorter = min(N, M) # N과 M중에 짧은 쪽을 판별
    longer = max(N, M) # N과 M중에 긴 쪽을 판별
    temp = []

    A = list(map(int, input().split()))
    B = list(map(int, input().split()))

    for j in range(longer - shorter + 1):
        sum = 0
        for i in range(shorter):
            sum += (A[i] * B[i])
        temp.append(sum)
        if len(A) > len(B):
            A.pop(0)
        else:
            B.pop(0)

    print("#{0} {1}".format(test_case, max(temp)))

profile
간절한 사람

0개의 댓글