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);
}
}
}