[SW Expert Academy] 1959번 두 개의 숫자열

문제

N 개의 숫자로 구성된 숫자열 Ai (i=1~N) 와 M 개의 숫자로 구성된 숫자열 Bj (j=1~M) 가 있다.
아래는 N =3 인 Ai 와 M = 5 인 Bj 의 예이다.

Ai 나 Bj 를 자유롭게 움직여서 숫자들이 서로 마주보는 위치를 변경할 수 있다.
단, 더 긴 쪽의 양끝을 벗어나서는 안 된다. 서로 마주보는 숫자들을 곱한 뒤 모두 더할 때 최댓값을 구하라.

 

입력

가장 첫 줄에는 테스트 케이스의 개수 T가 주어지고, 그 아래로 각 테스트 케이스가 주어진다.
각 테스트 케이스의 첫 번째 줄에 N 과 M 이 주어지고, 두 번째 줄에는 Ai,세 번째 줄에는 Bj 가 주어진다.

 

출력

출력의 각 줄은 '#t'로 시작하고, 공백을 한 칸 둔 다음 정답을 출력한다.

 

코드

import java.util.Scanner;
import java.io.FileInputStream;
 
class Solution
{
    public static void main(String args[]) throws Exception
    {
        Scanner sc = new Scanner(System.in);
        int t = sc.nextInt();
        
        for(int i=0; i<t; i++){
            int a = sc.nextInt();
            int b = sc.nextInt();
            int[] aArr = new int[a];
            int[] bArr = new int[b];
            
            for(int j=0; j<a; j++){ //a배열 값 받아오기
                aArr[j] = sc.nextInt();
            }
            
            for(int j=0; j<b; j++){ //b배열 값 받아오
                bArr[j] = sc.nextInt();
            }
            
            int max = 0;
           int[] big = (a>b)?aArr:bArr; //큰 배열
            int[] small = (a<b)?aArr:bArr; //작은 배열
             for(int j=0; j<big.length-small.length+1; j++){
                 int sum = 0;
                 for(int k=0; k<small.length; k++){
                    sum += big[j+k] * small[k];
                 }
                 max = (int)Math.max(sum,max);
            }
            System.out.printf("#%d %d\n",i+1,max);
        }
    }
}