[백준] 1026번: 보물

C++에서 벡터 sorting하기.

필요한 라이브러리는 #include<algorithm>

Sort는 다음과 같이 정의. sort(시작, 끝, 비교함수).

 

보통은 마지막을 생략하면(비교함수) 오름차순으로 정렬된다.

만약 내림차순으로 하고 싶다면 비교함수를 정의해주면 된다.

 

내가 짠 코드

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

bool desc(int a, int b){
    return a>b;
}

int main(void){
    int N;
    vector<int> A, B;
    int i,value,S=0;
    scanf("%d",&N);
    for(i=0;i<N;i++) {
        scanf("%d",&value);
        A.push_back(value);
    }
    for(i=0;i<N;i++) {
        scanf("%d",&value);
        B.push_back(value);
    }
    
    sort(A.begin(),A.end());
    sort(B.begin(),B.end(),desc);
    for(i=0;i<N;i++) S+= A[i]*B[i];
    cout << S << endl;
    return 0;
}