백준 - 1026번 보물(Greedy, Silver4, 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 오름차순

후 합 출력

---

변수

n : A, B 의 갯수
a : 입력 값 2
b = 입력 값 3

result : A B의 각 인덱스를 곱한 값을 더한 내용을 담을 변수

---

Sort

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

For-Loop 1.

A,B 두 리스트의 각 인덱스를 하나씩 순회하며 곱셈을 한 값을 누적합 시켜준다.

Print

A, B 의 곱셈 합이 들어있는 result 를 출력한다.

---

CODE

import sys

input = sys.stdin.readline

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

a.sort(reverse=True)
b.sort()

for i in range(n):
    result += a[i] * b[i]
print(result)