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

문제

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

프로그래머스

문제풀이

입출력 예 설명과, 몇 가지 테스트를 해봤더니 가장 작은수와 가장 큰수를 곱하여 더하는 값이 최솟값인걸 알았다.

따라서 A를 오름차순으로, B를 내림차순으로 하여 순서대로 곱한 값을 answer에 더해준다.

 

코드

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

int solution(vector<int> A, vector<int> B)
{
    int answer(0), idx(0);

    sort(A.begin(), A.end());
    sort(B.begin(), B.end(), greater<int>());

    for (idx = 0; idx < A.size(); ++idx)
        answer += A[idx] * B[idx];
    return answer;
}