[Programmers] Lv2. 최솟값 만들기 | C++

🔗문제 보러가기

프로그래머스

 

👨‍💻풀이 과정

 

자연수로만 이루어진 길이가 같은 두 배열이니, 하나는 오름차순으로, 다른 하나는 내림차순으로 정렬 후 각각 내적하면 되겠다 싶었습니다. 예를 들어, vector A = { \(a_1, a_2, a_3\) }, vector B = { \(b_1, b_2, b_3\) } 라고 한다면, 벡터의 내적(Dot product) = { \( a_1b_1 + a_2b_2 + a_3b_3\) }와 같습니다.

 

✏️소스 코드 및 결과

#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.begin(), B.end(), greater<int>());
    
    for(int i = 0; i < A.size(); i++)
        answer += (A[i] * B[i]);
    return answer;
}

 

 

 

몰랐는데, <numeric> 헤더에 두 시퀀스의 내적을 계산해주는 inner_product()란 함수가 있더군요.