BOJ 1026번: 보물

www.acmicpc.net/problem/1026

1026번: 보물

아이디어: 정렬

B의 가장 큰 수에 A의 가장 작은 수를 곱하면 된다.

A는 오름차순으로, B는 내림차순으로 정렬한 후 각각 곱한다.

5

1 1 1 6 0

2 7 8 3 1

이 예제에서 A를 오름차순으로 정렬하면 0 1 1 1 6, B를 내림차순으로 정렬하면 8 7 3 2 1 이므로 구하는 합은 0+7+3+2+6=18 이다.

 

코드:

파이썬에서 내림차순으로 정렬하는 방법, lambda를 쓰는 방법 잘 익혀두자!

#정렬
import sys
n = int(sys.stdin.readline())
A = list(map(int, sys.stdin.readline().split()))
B = list(map(int, sys.stdin.readline().split()))

A.sort() #오름차순 정렬
B.sort(reverse = True) #내림차순 정렬
ans = 0
for i in range(n):
    ans += A[i]*B[i]
print(ans)