[Codility] MaxProductOfThree - JavaScript

Sohyeon Bak·2021년 12월 7일

Codility

목록 보기

11/19

문제

A non-empty array A consisting of N integers is given. The product of triplet (P, Q, R) equates to A[P] A[Q] A[R] (0 ≤ P < Q < R < N).

For example, array A such that:

A[0] = -3
A[1] = 1
A[2] = 2
A[3] = -2
A[4] = 5
A[5] = 6
contains the following example triplets:

(0, 1, 2), product is −3 1 2 = −6
(1, 2, 4), product is 1 2 5 = 10
(2, 4, 5), product is 2 5 6 = 60
Your goal is to find the maximal product of any triplet.

Write a function:

function solution(A);

that, given a non-empty array A, returns the value of the maximal product of any triplet.

For example, given array A such that:

A[0] = -3
A[1] = 1
A[2] = 2
A[3] = -2
A[4] = 5
A[5] = 6
the function should return 60, as the product of triplet (2, 4, 5) is maximal.

Write an efficient algorithm for the following assumptions:

N is an integer within the range [3..100,000];
each element of array A is an integer within the range [−1,000..1,000].

문제해석

배열 A 중 요소 3개의 곱 중 가장 큰 값을 리턴하는 문제이다.
sort를 이용해 문제를 해결할 수 있는데, 문제는 값을 곱하는 것이기 때문에 음수 2개를 곱한 값은 양수가 되어 값의 크기가 변하게 된다. 그 점을 유의하여 문제를 풀 수 있어야 한다.

문제풀이

Sort를 이용하여 오름차순으로 배열 A를 정렬한다.
가장 큰 값을 구할 수 있는 방법은 정렬된 배열 A의 가장 마지막 세 수의 곱과 배열의 첫번째 두번째 수와 그리고 마지막 수를 곱한 값이다.
- 배열을 오름차순으로 정렬할때 요소 중 가장 작은 두 값이 음수라면 두 값을 곱했을때 정수가 되고, 그 값과 배열에서 가장 큰 값을 곱하면 배열 중 큰 값이 나오기 때문에 두개의 값을 도출해 내야한다.
찾은 두 개의 값을 비교해 가장 큰 값을 답으로 지정한다.

코드

function solution(A) {
    // write your code in JavaScript (Node.js 8.9.4)
    let answer = 0;
    A.sort((a,b)=>a-b)
    
    let a = A[A.length-1] * A[A.length-2] * A[A.length-3];
    let b = A[0] * A[1] * A[A.length-1];
    answer = Math.max(a, b);
    
    return answer 
}