[알고리즘] BOJ 16208 귀찮음

[BOJ] 16208 귀찮음 바로가기

📍 문제

현우는 무슨 이유에선지 길이 a1, ..., an의, 총 n개의 쇠막대가 필요해졌다. 하지만 그가 가진 것은 길이 a1+...+an의 하나의 쇠막대뿐이었다. 현우는 이 막대를 직접 잘라서 원래 필요하던 n개의 쇠막대를 만들 것이다. 길이 x+y인 막대를 길이 x, y인 두 개의 막대로 자를 때에는 만들려 하는 두 막대의 길이의 곱인 xy의 비용이 든다. 현우는 최소의 비용으로 이 쇠막대를 잘라서 a1, ..., an의 n개의 쇠막대를 얻고 싶다.

그런데 현우는 이 비용이 얼마나 들지 잘 모르겠다. 그래서 여러분이 막대를 자르는 최소 비용을 계산하는 프로그램을 작성해주면 코드잼 경시대회 점수를 30점 올려주겠다고 제안했다. 어떤가?

---

📍 입력

첫째 줄에는 현우가 원하는 쇠막대의 수를 나타내는 정수 n이 주어진다. (1 ≤ n ≤ 500,000)

둘째 줄에는 현우가 원하는 쇠막대의 길이를 나타내는 정수 a1, ..., an이 주어진다. (1 ≤ ai ≤ 101)

---

📍 출력

현우가 필요한 n개의 쇠막대를 얻는 최소의 비용을 출력한다.

---

📍 풀이

✍ 코드

from sys import stdin
n = int(stdin.readline())
A = sorted(list(map(int,stdin.readline().split()))) # 파이프 길이 오름차순 정렬
total = sum(A) # 파이프의 총 길이
result = 0
for a in A:
  total -= a # 얻어야 하는 파이프 길이만큼 잘라낸 길이 
  result += a * total
print(result)