[코딩테스트] BJ1026 보물 - C++, PYTHON

✔문제

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

길이가 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의 최솟값을 출력한다.

💖C++ 코드

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std; 

int main() {

	int n; 
	int sum = 0;
	cin >> n; 

	vector <int> A; 
	vector <int> B; 

	for (int i = 0; i < n; i++) {
		int temp; 
		cin >> temp; 
		A.push_back(temp);
	}

	for (int i = 0; i < n; i++) {
		int temp;
		cin >> temp;
		B.push_back(temp);
	}

	sort(B.begin(), B.end());
	sort(A.begin(), A.end());

	for (int i = 0; i < n; i++) {
		sum += A[i] * B[n - i-1];
	}


	cout << sum;

	return 0;
}

💖파이썬 코드

n = int(input())
A = [] 
B = [] 
A = sorted(map(int, input().split(' '))) 
B = sorted(map(int, input().split(' ')))
sum = 0 



for i in range(n):
    sum += A[i]*B[n-i-1]

print(sum)