[SW expert Academy] SWEA 1959번 두 개의 숫자열 자바(Java)

@notepad_jj2

츄르사려고 코딩하는 코집사입니다.

---

1. [SW expert Academy] SWEA 1959번 두 개의 숫자열 자바(Java)

1) 난이도 : D2

 

2) 문제번호 : 1959번

 

3) 문제 출처

swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV5PpoFaAS4DFAUq&categoryId=AV5PpoFaAS4DFAUq&categoryType=CODE&problemTitle=1959&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=&pageSize=10&pageIndex=1#none

SW Expert Academy

 

2. 문제

N 개의 숫자로 구성된 숫자열 Ai (i=1~N) 와 M 개의 숫자로 구성된 숫자열 Bj (j=1~M) 가 있다.

아래는 N =3 인 Ai 와 M = 5 인 Bj 의 예이다.

Ai 나 Bj 를 자유롭게 움직여서 숫자들이 서로 마주보는 위치를 변경할 수 있다.

단, 더 긴 쪽의 양끝을 벗어나서는 안 된다.

서로 마주보는 숫자들을 곱한 뒤 모두 더할 때 최댓값을 구하라.

위 예제의 정답은 아래와 같이 30 이 된다.
 

3. 제약사항

- N 과 M은 3 이상 20 이하이다.

 

4. 입력

가장 첫 줄에는 테스트 케이스의 개수 T가 주어지고, 그 아래로 각 테스트 케이스가 주어진다.

각 테스트 케이스의 첫 번째 줄에 N 과 M 이 주어지고,

두 번째 줄에는 Ai,

세 번째 줄에는 Bj 가 주어진다.

 

5. 출력

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

(t는 테스트 케이스의 번호를 의미하며 1부터 시작한다.)

6. 풀이

- N과 M의 크기에 따라 풀이 방법이 달라짐

- N이 작을 경우에는 그림처럼 진행하면 되는데,

- M이 작을 경우에는 N의 반대로 하면 된다.

7. 소스 코드

import java.util.*;
 
import javax.print.attribute.IntegerSyntax;
 
public class Solution {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        //테스트 케이스 개수 입력
        int T = sc.nextInt();
         
        for(int tc=1;tc<=T;tc++) {
            //A 길이
            int N = sc.nextInt();
             
            //B 길이
            int M = sc.nextInt();
             
            //배열 선언
            int[] A = new int[N];
            int[] B = new int[M];
             
            for(int i=0;i<A.length;i++) A[i] = sc.nextInt();
            for(int i=0;i<B.length;i++) B[i] = sc.nextInt();
             
            //최대값
            int Max = 0;
             
            if(N<M) {
                for(int i=0;i<M-N+1;i++) {
                    int result = 0;
                    for(int j=0;j<N;j++) {
                        result += A[j] *  B[i+j];
                    }
                    Max = Math.max(Max, result);
                }
            }
             
            if(N>M) {
                for(int i=0;i<N-M+1;i++) {
                    int result = 0;
                    for(int j=0;j<M;j++) {
                        result += A[i+j] *  B[j];
                    }
                    Max = Math.max(Max, result);
                }
            }
            else {
                 
                int result=0;
                 
                for(int a=0;a<N;a++) {
                    result += A[a] * B[a];
                }
                Max = Math.max(Max, result);
            }
             
             
            System.out.printf("#%d %d\n",tc,Max);
        }
         
         
         
         
    }   
}

---