안녕 세상아,

[c++/프로그래머스] LV2 최솟값 만들기 본문

프로그래머스

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

돈 많은 백수가 되고싶다 2023. 5. 8. 19:52

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

(가장 작은 값 * 가장 큰 값) + (두번째로 작은 값 * 두번째로 큰 값) + ... + (가장 큰 값 * 가장 작은 값)

일케 구하면 된다. 

이렇게 하기 위해 두 벡터를 모두 정렬하면 된다.

 

sort 정렬은 기본적으로 오름차순으로 되어있기 때문에 greater<>()를 sort 맨 뒤에 붙여주면 내림차순 정렬로 바뀐다.

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

int solution(vector<int> A, vector<int> B)
{
    int answer = 0;
    int vsize = A.size();

    sort(A.begin(), A.end());				//오름차순 정렬
    sort(B.begin(), B.end(), greater<>());	//내림차순 정렬
    
    for (int i = 0; i < vsize; i++) {
        answer = answer + (A[i] * B[i]);
    }
    return answer;
}
int main() {
    vector<int> a;
    vector<int> b;

   
    cout << solution(a, b);
}