[백준] 1026번 보물 - C++

🌱 문제 설명

1026번 보물

옛날 옛적에 수학이 항상 큰 골칫거리였던 나라가 있었다. 이 나라의 국왕 김지민은 다음과 같은 문제를 내고 큰 상금을 걸었다.

길이가 N인 정수 배열 A와 B가 있다. 다음과 같이 함수 S를 정의하자.

S = A[0] × B[0] + ... + A[N-1] × B[N-1]

S의 값을 가장 작게 만들기 위해 A의 수를 재배열하자. 단, B에 있는 수는 재배열하면 안 된다.

S의 최솟값을 출력하는 프로그램을 작성하시오.

---

입력

첫째 줄에 N이 주어진다. 둘째 줄에는 A에 있는 N개의 수가 순서대로 주어지고, 셋째 줄에는 B에 있는 수가 순서대로 주어진다. N은 50보다 작거나 같은 자연수이고, A와 B의 각 원소는 100보다 작거나 같은 음이 아닌 정수이다.

---

출력

첫째 줄에 S의 최솟값을 출력한다.

---

풀이

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

vector<int> A;
vector<int> B;
int N;
int answer = 0;

bool compare(int x, int y) {
    return x > y;
}

int main() {
    int num;
    cin >> N;
    for (int i = 0; i < N; i++) {
        cin >> num;
        A.push_back(num);
    }
    for (int i = 0; i < N; i++) {
        cin >> num;
        B.push_back(num);
    }
    sort(A.begin(), A.end());//올림차순 정렬
    sort(B.begin(), B.end(), compare);//내림차순 정렬

    for (int i = 0; i < N; i++) {
        answer += (A[i] * B[i]);
    }
    cout << answer << endl;
    return 0;
}

❣️문제 요약:

2개의 배열 A,B에 수를 입력받아서 A,B의 인덱스 곱의 총 합이 최소가 되도록 A 배열을 재배열 한다.

🤔 생각해야 할 점

1. 같은 인덱스의 배열 수 끼리 곱해서 더한 총 합의 최소를 구한다.
2. A배열의 가장 큰 수를 B 배열의 가장 작은 수와 곱한다면 최소값을 구할 수 있다.

→ A배열은 오름차순으로 정렬하고 B배열은 내림차순 정렬하면 두 인덱스 곱의 최소가 된다.

✅ vector 클래스를 사용해서 sort( vector.end(), vector.begin() ) 오름차순 정렬

코드 설명

✅ 내림차순으로 정렬하는데 쓰이는 함수 compare

bool compare(int x, int y) {
    return x > y;
}

✅ A배열과 B 배열 입력 받기

 for (int i = 0; i < N; i++) {
        cin >> num;
        A.push_back(num);//벡터배열 뒤로 입력받은 숫자 추가
    }
    for (int i = 0; i < N; i++) {
        cin >> num;
        B.push_back(num);
    }

✅ A벡터는 오름차순으로 정렬하고 B벡터는 compare함수를 이용해 내림차순 정렬

sort(A.begin(), A.end());//올림차순 정렬
sort(B.begin(), B.end(), compare);//내림차순 정렬

✅A벡터와 B벡터의 곱을 answer 변수에 저장

for (int i = 0; i < N; i++) {
        answer += (A[i] * B[i]);
    }

✅ 정렬

1. 일반 배열

   오름차순: sort(arr, arr+배열범위)

   내림차순: sort(arr+배열범위, arr)

2. 벡터 배열

   오름차순: sort(v.begin(), v.end())

   내림차순: sort(v.rbegin(), v.rend());

아니면 내림차순의 경우 compare함수를 만들어서

bool compare(int x,int y){
return x < y;
}
...
sort(B.begin(), B.end(),compare);