[프로그래머스/C++] 최솟값 만들기

1. 문제 개요

https://programmers.co.kr/learn/courses/30/lessons/12941

코딩테스트 연습 - 최솟값 만들기

1-1. 설명

길이가 동일한 배열 A, B가 있다. 각 배열은 자연수로 존재한다. 각 배열에서 한개의 숫자를 뽑아 두 수를 곱한다. 이러한 과정을 배열 길이만큼 반복해 곱한 값을 누적해 더한다. 이때 최종적으로 누적된 값이 최소가 되도록 만드는 것이 목표이다. 배열 A, B가 주어질 때 최종적으로 누적된 최솟값을 return하는 solution 함수를 작성한다. 

 

1-2. 제한사항

 - 배열 A, B의 크기는 1,000 이하의 자연수

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

 

---

2. 구현

2-1. 풀이

곱한 값의 합이 최솟값이 되어야 하므로, A는 오름차순으로 B는 내림차순으로 정렬하여 각각의 원소를 뽑아 곱해주었다. 

 

2-2. 코드

#include <vector>
#include <algorithm>
using namespace std;

int solution(vector<int> A, vector<int> B)
{
    int answer = 0;
    sort(A.begin(), A.end());
    sort(B.rbegin(), B.rend());
    
    for (int i = 0; i < B.size();i++) answer += (A[i] * B[i]);
    
    return answer;
}