알고리즘(C++)/Programmers 알고리즘
최솟값 만들기
개발하는꼬물이
2017. 10. 2. 20:37
문제. 자연수로 이루어진 길이가 같은 수열 A,B가 있습니다. 최솟값 만들기는 A, B에서 각각 한 개의 숫자를 뽑아 두 수를 곱한 값을 누적하여 더합니다. 이러한 과정을 수열의 길이만큼 반복하여 최종적으로 누적된 값이 최소가 되도록 만드는 것이 목표입니다.
예를 들어 A = [1, 2]
, B = [3, 4]
라면
1. A에서 1, B에서 4를 뽑아 곱하여 더합니다.
2. A에서 2, B에서 3을 뽑아 곱하여 더합니다.
수열의 길이만큼 반복하여 최솟값 10을 얻을 수 있으며, 이 10이 최솟값이 됩니다.
수열 A,B가 주어질 때, 최솟값을 반환해주는 getMinSum 함수를 완성하세요.
#include <iostream> #include <vector> #include <algorithm> using namespace std; int getMinSum(vector<int> A, vector<int> B) { int answer = 0; vector<int> temp; // A는 오름차순, B는 내림차순으로 정렬 sort(A.begin(), A.end()); sort(B.rbegin(), B.rend()); for (int i = 0; i < A.size(); i++) { answer += A[i] * B[i]; } return answer; } int main() { vector<int> tA{ 1, 2 }, tB{ 3, 4 }; //아래는 테스트 출력을 위한 코드입니다. cout << getMinSum(tA, tB); }
-> algorithm 헤더에 있는 std::sort를 이용하면 쉽게 풀 수 있음!
sort(A.begin(), A.end()); // A 벡터의 처음부터 끝까지 오름차순으로 정렬
// begin, end 앞에 r을 붙이면 내림차순으로 정렬이 가능!