[Beakjoon] 1026 : 보물 - 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의 최솟값을 출력한다.

---

풀이

일단 해당 문제를 해결하는 방법은 A의 가장 작은 숫자와 B의 가장 큰 숫자를 곱하는 것이다.

이 후에는 A의 두번째로 작은 숫자와 B의 두번째로 작은 숫자를 곱하고.... 반복해서 더해간다.

조건에 B는 재배열 할 수 없다고 나와 있으므로 sort()를 사용하는 방법 보다는 min()과 max()를 활용하는 것으로 결정했다.

추가로 배열을 재배열 하면 안된다는 조건이 있으니 remove를 통해 계산이 완료 된 요소는 제거하는 방식으로 구현했다.

 

n = int(input())
a = list(map(int, input().split()))
b = list(map(int, input().split()))

result = 0

for i in range(n):
    min_num = min(a)
    max_num = max(b)
    
    a.remove(min_num)
    b.remove(max_num)
    
    result += min_num * max_num
    
print(result)