티스토리 뷰

알고리즘(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을 붙이면 내림차순으로 정렬이 가능!


'알고리즘(C++) > Programmers 알고리즘' 카테고리의 다른 글

2016 날짜 계산  (0) 2017.10.02
최대공약수와 최소공배수  (0) 2017.10.02
소수 찾기  (0) 2017.10.02
콜라츠 추측  (0) 2017.10.02
행렬의 덧셈  (0) 2017.09.28
댓글