Level 2 - 최솟값 만들기 (js sort 제대로 사용하기)

문제 설명

길이가 같은 배열 A, B 두개가 있습니다. 각 배열은 자연수로 이루어져 있습니다.
배열 A, B에서 각각 한 개의 숫자를 뽑아 두 수를 곱합니다. 이러한 과정을 배열의 길이만큼 반복하며, 두 수를 곱한 값을 누적하여 더합니다. 이때 최종적으로 누적된 값이 최소가 되도록 만드는 것이 목표입니다. (단, 각 배열에서 k번째 숫자를 뽑았다면 다음에 k번째 숫자는 다시 뽑을 수 없습니다.)

예를 들어 A = [1, 4, 2] , B = [5, 4, 4] 라면

• A에서 첫번째 숫자인 1, B에서 첫번째 숫자인 5를 뽑아 곱하여 더합니다. (누적된 값 : 0 + 5(1x5) = 5)
• A에서 두번째 숫자인 4, B에서 세번째 숫자인 4를 뽑아 곱하여 더합니다. (누적된 값 : 5 + 16(4x4) = 21)
• A에서 세번째 숫자인 2, B에서 두번째 숫자인 4를 뽑아 곱하여 더합니다. (누적된 값 : 21 + 8(2x4) = 29)

즉, 이 경우가 최소가 되므로 29를 return 합니다.

배열 A, B가 주어질 때 최종적으로 누적된 최솟값을 return 하는 solution 함수를 완성해 주세요.

제한 사항

• 배열 A, B의 크기 : 1,000 이하의 자연수
• 배열 A, B의 원소의 크기 : 1,000 이하의 자연수

입출력 예

입출력 예 설명

입출력 예 #1
문제의 예시와 같습니다.

 

입출력 예 #2
A에서 첫번째 숫자인 1, B에서 두번째 숫자인 4를 뽑아 곱하여 더합니다. (누적된 값 : 4) 다음, A에서 두번째 숫자인 2, B에서 첫번째 숫자인 3을 뽑아 곱하여 더합니다. (누적된 값 : 4 + 6 = 10)
이 경우가 최소이므로 10을 return 합니다.

---

이 문제의 key포인트

-> sort... sort 주의

 

---

자료구조 및 알고리즘: 그리디

---

풀이

1차시도 (sort 문제)

function solution(A,B){
    let result = 0;
    A.sort(); // 오름차순 정렬
    
    // 내림차순 정렬 
    B.sort(); 
    B.reverse();

	// 최소누적합 구하기
    result = A.reduce((acc, n, idx) => {
        console.log(acc, n, idx)
      return acc += n * B[idx];
    }, 0)
    
    return result;
}

문제를 풀 때 아이디어가 떠오르지 않아서 Q&A를 슬쩍봤을 때 sort라는 힌트를 얻었다. 

A배열에서 값을 먼저 선택한다고 생각하면 A배열의 가장 작은 수와 B배열의 가장 큰 수를 곱하게 하면 결론적으로 누적합이 제일 작아진다는 생각을 했다.

따라서 힌트에서 얻은 sort를 이용해 A배열은 오름차순, B배열은 내림차순 정렬을 해서 동일한 인덱스 값을 곱한걸 누적하도록 구현했다. 그런데,, 샘플 테케와 테케 1번만 통과하고 나머지는 실패가 주루룩 떴다. 

(처음에는 sort가 문제인지 모르고 이 방법이 아닌가 해서 브루트포스로 풀다가 포기했다.......)

 

문제는 내가 sort를 돌리는 방식이었다!!!!

자바스크립트에서 sort는 모든 요소를 문자열로 인식해서 이 문자열을 정렬한다.

(파이썬에서는 그냥 무작정 sort돌리면 정수로 정렬해준다..)

그래서 다음과 같은 문제점이 생긴다.

let A = [5, 2, 13, 42]
console.log(A.sort()) // [13, 2, 42, 5]

13을 문자열로 인식하기 때문에 앞의 "1" 과 "2", "4", "5" 를 비교하면 13이 제일 작다고 판단하고 첫번째로 정렬되는 것이다. 우리가 예상한 [2, 5, 13, 42]와는 아주 다른 결과다.

 

따라서 배열을 오름차순 정렬하려면 sort() 함수에 값을 비교해줄 수 있는 함수를 넣어줘야한다. 
두 숫자의 차이를 비교해서 값이 양수면 양수를, 같으면 0을, 음수면 음수를 return하는 함수이다.

오름차순 정렬

let A = [5, 2, 13, 42]
A.sort((a, b) => a - b);
console.log(A) // -> [2, 5, 13, 42]

 내림차순 정렬

let A = [5, 2, 13, 42]
A.sort((a, b) => b - a);
console.log(A) // -> [42, 13, 5, 2]

 

 

완성코드

function solution(A,B){
    let result = 0;
    A.sort((a, b) => a - b);
    B.sort((a, b) => b - a);

    result = A.reduce((acc, n, idx) => {
      return acc += n * B[idx];
    }, 0)
    
    return result;
}