[SWEA] D2 1959. 두 개의 숫자열 _Java

 

 

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

 

입력	출력
10 3 5 1 5 3 3 6 -7 5 4 7 6 6 0 5 5 -1 1 6 -4 1 8 7 -9 3 ...	#1 30 #2 63 ...

 

 

 

🧶 풀이법

1. 배열의 값을 각각 입력받는다.

2. 각 배열의 길이를 비교하고 짧은 쪽을 기준으로 계산식을 수행한다.

3. 각 요소를 곱하고 더하는 계산을 짧은 배열의 길이만큼 반복수행하되, 한 사이클의 계산이 끝나면 인덱스를 +1 해주고 다시 계산을 반복한다.

4. 각 계산의 결과값을 result에 저장해두고 sum 결과값과 비교했을때 더 클 경우 값을 업데이트한다.

 

🧶 코드

import java.util.Scanner;
import java.io.FileInputStream;
import java.util.Arrays;

class Solution
{
	public static void main(String args[]) throws Exception
	{

		Scanner sc = new Scanner(System.in);
		int T;
		T=sc.nextInt();
 
		for(int test_case = 1; test_case <= T; test_case++)
		{
            // 각 배열의 길이
            int lenA = sc.nextInt();
            int lenB = sc.nextInt();
           	
            // 각 배열의 길이만큼 배열 생성 후 값 입력받기
            int[] arrA = new int[lenA];
            for(int a = 0; a < lenA; a++){
                arrA[a] = sc.nextInt();
            }         
            int[] arrB = new int[lenB];
            for(int b = 0; b < lenB; b++){
                arrB[b] = sc.nextInt();
            }

            int result = 0;
            int s = 0;

            // A배열과 B배열의 길이의 차이만큼 반복 수행
            for(int i = 0; i < Math.abs(lenA - lenB) + 1; i++){
                int len = 0;
                int sum = 0;
				if(lenA > lenB) { // 배열A가 더 길 경우
                    len = lenB; // 더 짧은 쪽의 배열의 길이만큼 계산식 반복 수행
                    for(int j = 0; j < len; j++){
                        sum += arrA[j+s]*arrB[j];
                    }
                    s++; // 짧은 쪽의 배열 인덱스 한 칸씩 오른쪽으로 옮김
                }else{  // 배열B가 더 길 경우
                    len = lenA;  // 더 짧은 쪽의 배열의 길이만큼 계산식 반복 수행
                    for(int j = 0; j < len; j++){
                        sum += arrA[j]*arrB[j+s];
                    }
                    s++;
                }
                // sum에 더 큰 합 넣기
                if(sum > result) {
	                result = sum;
                }
            }
           
            System.out.println("#" + test_case + " " + result);
        }
	}
}

 

 

🧶 후기

풀었던걸 적으려니 기억이 잘 안나네요 다음엔 풀자마자 기록해야겠어요 👀