Programmers - 최솟값 구하기

프로그래머스

 

문제

길이가 같은 배열 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 이하의 자연수

 

 

접근방식

두 배열의 값들을 비교하고 각각 값을 곱하고 최종적으로 모든 값을 더해서 가장 작은 값을 도출하는 문제

가장 작은 값이 어떻게하면 나올까 생각해보았는데, 배열 A를 오름차순으로 정렬하고 배열 B를 내림차순으로 정렬해서 각각의 인덱스의 값들끼리 곱해주면, 큰 수의 값이 커지는 것을 막을 수 있으므로 가장 작은 값이 나오게 되는 것을 알게 됨

 

따라서 두 배열 중 하나는 오름차순으로, 하나는 내림차순으로 정렬한 후 파이썬의 내장 함수인 zip 함수를 써서 두 배열에서 값을 하나씩 꺼내 곱해주고 ans_arr 안에 선언해준 후, ans_arr의 합을 return 해줌

코드

def solution(A,B):
    A.sort()
    B.sort(reverse=True)
    
    ans_arr = [x * y for x, y in zip(A,B)]
        
    return sum(ans_arr)

 

 

lv2 문제가 맞나..?