[python] 1026, 보물

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

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

 

[예제]

 

[풀이]

• 정렬 사용

n = int(input())
a = list(map(int, input().split()))
b = list(map(int, input().split()))
a.sort()
b.sort(reverse = True)
result = 0

for aa, bb in zip(a, b):
  result += aa * bb

print(result)

• 정렬 미사용

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

for i in a:
  result += i * b.pop(b.index(max(b)))

print(result)

• a의 리스트는 정렬
• b의 리스트는 정렬이 불가하기 때문에 정렬하지 않고 가장 큰 값을 a의 리스트에 순서대로 곱하기
• 곱할 때마다 b 리스트의 최대값을 제거해주며 다음번에 곱할 때는 그 다음으로 큰 수를 곱할 수 있도록