[SWEA] 1959 : 두개의 숫자열 - Java

문제

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

import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

class Solution
{
    static int n,m;
    static int[] dataA, dataB;
    public static void main(String args[]) throws Exception
    {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        int T;
        T = Integer.parseInt(br.readLine());
        for(int test_case = 1; test_case <= T; test_case++)
        {
            st = new StringTokenizer(br.readLine());
            n = Integer.parseInt(st.nextToken());
            m = Integer.parseInt(st.nextToken());
            dataA = new int[n];
            dataB = new int[m];
            st = new StringTokenizer(br.readLine());
            for (int i = 0; i < n; i++) {
                dataA[i] = Integer.parseInt(st.nextToken());
            }
            st = new StringTokenizer(br.readLine());
            for (int i = 0; i < m; i++) {
                dataB[i] = Integer.parseInt(st.nextToken());
            }
            int res = maxResult();
            System.out.println("#" + test_case + " " + res);
        }
    }

    public static int maxResult() {
        int res = 0;
        if (n < m) {
            for (int i = 0; i < m - n + 1; i++) {
                int sum = 0;
                for (int j = 0; j < n; j++) {
                    sum += dataA[j] * dataB[i+j];  //B의 더해지는 위치 조정.
                }
                res = Math.max(res, sum);
            }
        } else if (n > m) {
            for (int i = 0; i < n - m + 1; i++) {
                int sum = 0;
                for (int j = 0; j < m; j++) {
                    sum += dataA[i + j] * dataB[j];
                }
                res = Math.max(res, sum);
            }
        } else {
            int sum = 0;
            for (int i = 0; i < n; i++) {
                sum += dataA[i] * dataB[i];
            }
            res = Math.max(res, sum);
        }
        return res;
    }


}