SWEA 1959번: 두 개의 숫자열

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

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#define MAXSIZE 20
#define COMPARE(x, y)(x>y?x:y)

int calculate(int*, int*, int, int);

int main() {
	int T, N, M;
	int i, k;
	int A[MAXSIZE] = { 0, }, B[MAXSIZE] = { 0, };

	scanf("%d", &T);
	for (i = 0; i < T; i++) {
		scanf("%d %d", &N, &M);
		for(k=0;k<N;k++)
			scanf("%d", &A[k]);
		for(k=0;k<M;k++)
			scanf("%d", &B[k]);
		if (N > M)
			printf("#%d %d\n", i + 1, calculate(A, B, N, M));
		else
			printf("#%d %d\n", i + 1, calculate(B, A, M, N));
	}
	return 0;
}

int calculate(int*big, int*small, int l_b, int l_s) {
	int max_result = 0, result = 0;
	int i, j;
	for (i = 0; i <= l_b - l_s; i++) {
		for (j = 0; j < l_s; j++) {
			result += big[i + j] * small[j];
		}
		max_result = COMPARE(max_result, result);
		result = 0;
	}
	return max_result;
}

채점 결과