[baekjoon] 1026 : 보물 (C++)

문제 

---

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

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

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

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

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

 

입력

---

첫째 줄에 N이 주어진다. 둘째 줄에는 A에 있는 N개의 수가 순서대로 주어지고, 셋째 줄에는 B에 있는 수가 순서대로 주어진다.

N은 50보다 작거나 같은 자연수이고, A와 B의 각 원소는 100보다 작거나 같은 음이 아닌 정수이다.

 

출력

---

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

 

예제 입출력

---

https://www.acmicpc.net/problem/1026

1026번: 보물

 

문제 풀이

---

먼저 S의 값을 가장 작게 만들기 위해서는 A의 가장 작은수와 B의 가장 큰수를 곱하여 더하면 된다.

하지만 B에 있는 수는 재배열 하면 안된다는 조건이 있으니 A만 오름차순으로 재배열 후 B에서 가장 큰수부터 뽑아와 A의 첫번째 수와 곱해준 후 B의 가장 큰 수를 제거해주면 된다.

 

오류 코드

---

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

int main() {
	int n, j;
	vector<int> A;
	vector<int> B;
	int m;
	int sum = 0;
	int max = 0;
	cin >> n;
	m = n;
	for (int i = 0; i < n; i++) {
		cin >> j;
		A.push_back(j);
	}
	for (int i = 0; i < n; i++) {
		cin >> j;
		B.push_back(j);
	}
	sort(A.begin(), A.end());
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < m; j++) {
			if (max < B[j])
				max = B[j];
		}
		sum += max * A[i];
		int k = find(B.begin(), B.end(), max) - B.begin();
		B.erase(B.begin() + k);
		m--;
	}
	cout << sum;
}

원래는 B에서 제일 큰수를 뽑고 그수의 인덱스 번호를 찾은 후 제거하는 방식으로 아래와 같이 선택했었는데 무슨이유인지 오류가 떠서 다른 비슷한 방법을 선택하였다.

 

최종 코드

---

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

int main() {
	int n, j;
	vector<int> A;
	vector<int> B;
	int m;
	int sum = 0;
	int max = 0;
	cin >> n;
	m = n;
	for (int i = 0; i < n; i++) {
		cin >> j;
		A.push_back(j);
	}
	for (int i = 0; i < n; i++) {
		cin >> j;
		B.push_back(j);
	}
	sort(A.begin(), A.end());
	
	for (int i = 0; i < n; i++) {
		int maxB = *max_element(B.begin(), B.end());
		sum += A[i] * maxB;
		int k = find(B.begin(), B.end(), maxB) - B.begin();
		B.erase(B.begin() + k);
	}
	cout << sum;
}