Categories: SWEA

SWEA 1959 두 개의 숫자열 D2 (자바 Java)

java

SWEA 1959 두 개의 숫자열

문제 정보

문제 링크

문제 출처 : SW Expert Academy
문제 번호 : 1959
문제 제목 : 두 개의 숫자열
문제 난이도 : D2
풀이 언어 : Java

문제

N과 M의 길이로 된 길이가 서로 다른 숫자열 A 와 B 가 주어진다.
A 와 B를 나란히 놓고 짧은 쪽을 한칸씩 움직이며 마주 보는 숫자들을 곱해서 더한 수 중에 최대값을 구하는 문제

접근

N의 길이가 길 때와 M의 길이가 길 때를 따로 분리해주고, 2중 for 문을 돌면서 변수 result 에 곱한 숫자의 합을 넣고 그 중 최대값을 구했다.

풀이

첫 줄에 주어지는 숫자 T는 test_case 개수이다.
각 테스트 케이스 첫 줄에는 숫자 N 과 M이 주어진다.
그 다음 줄에는 A배열의 숫자들, 또 그 다음 줄에는 B배열의 숫자들이 주어진다.
max 변수를 초기화 한 후
N이 M보다 작은 경우 배열 A를 움직이면서 배열 B에 마주 보는 숫자들을 곱한 후 더해준다.
N이 M보다 큰 경우 배열 B를 움직이면서 배열 A에 마주 보는 숫자들을 곱한 후 더해준다.
N과 M이 같은 경우 두 배열이 마주 보는 숫자들을 곱한 후 더해준다.
Math.max() 를 사용해 최대값을 변수 max 에 갱신해준다.

후기

각 배열의 인덱스 위치가 헷갈릴 수 있는 문제입니다. 직접 그려보면서 하면 좀 더 빠르게 이해할 수 있습니다.

코드

Java

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;
        T=sc.nextInt();
         
        for(int test_case = 1; test_case <= T; test_case++)
        {
            int N = sc.nextInt();
            int M = sc.nextInt();
             
            int[] A = new int[N];
            int[] B = new int[M];
             
            for(int i = 0; i < N; i++) A[i] = sc.nextInt();
            for(int i = 0; i < M; 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);
                }
            } else 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 i = 0; i < N; i++) {
                    result += A[i]*B[i];
                }
                max = Math.max(max, result);
            }
            System.out.printf("#%d %d\n", test_case, max);
        }
    }
}