[파이썬/알고리즘] 프로그래머스 - 최솟값 만들기

[문제]

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

(단, 각 배열에서 k번째 숫자를 뽑았다면 다음에 k번째 숫자는 다시 뽑을 수 없습니다.)

 

[풀이]

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

 

양 쪽 배열을 곱해서 최솟값이 나오려면 A 배열은 내림차순, B 배열은 오름차순으로 정렬한다.

그리고 각 배열을 순서대로 곱해주면 된다.

 

배열에 대해서 길이만큼 반복문을 돌리려면 range(len(배열)) 을 알아야한다.

배열.sort() 하면 오름차순으로 정렬된다. Ex : [2, 3, 4, 5]

배열.sort(reverse=True) 하면 내림차순으로 정렬된다. [5, 4, 3, 2]