[연습문제] 최솟값 만들기 - Python/JavaScript

문제 링크 - https://school.programmers.co.kr/learn/courses/30/lessons/12941

프로그래머스

 

[문제 간단 요약]

1. 길이가 같은 배열에서 하나씩 곱하여 누적 합이 최대인 값을 구하는 문제

[문제 해결 방안]

1. 배열 2개를 하나는 오름차순 하나는 내림차순으로 정렬하여 순차적으로 하나씩 선택하여 곱한 누적합을 통해 문제 해결

 

[문제 해결 코드 - Python]

def solution(A,B):
    answer = 0
    A.sort()
    B.sort(reverse=True)
    
    for i in range(len(A)):
        answer += A[i]*B[i]

    return answer

 

[문제 해결 코드 - JavaScript]

function solution(A,B){
    var answer = 0;
    A.sort((a,b) => a-b)
    B.sort((a,b) => b-a)
    for (i=0; i<=A.length-1; i++) {
        answer = answer + A[i]*B[i]
    }

    return answer;
}

 

 

[개선점] for 문을 통해 곱을 구할 수도 있지만 reduce를 통해 누적합을 구할 수 있음

• reduce - 배열의 요소에 대해 주어진 함수를 실행하여, 하나의 결과 값을 제공
• array.reduce(Callback 함수 ,초기 값)

function solution(A,B){
    var answer = 0;
    A.sort((a,b) => a-b)
    B.sort((a,b) => b-a)
    return A.reduce((total, A_num, idx) => total+A_num*B[idx],0)
}