파이썬 | 백준 | 1715 | 카드 정렬하기

solution

1. 카드 묶음 배열 정렬

# 1715, 카드 정렬하기
import sys

n = int(sys.stdin.readline())
ns = []
for _ in range(n):
    ns.append(int(sys.stdin.readline()))

ns.sort()
temp = 0
for i in range(len(ns)-1):
    temp += ns[i] + ns[i+1]
    ns[i+1] = temp
print(temp)

메모리 초과

---

2. 힙 이용

# 1715, 카드 정렬하기
import sys
import heapq


n = int(sys.stdin.readline())
ns = []
for _ in range(n):
    heapq.heappush(ns, int(sys.stdin.readline()))

if len(ns) == 1:
    print(0)
else:
    ans = 0
    while len(ns) > 1:
        temp1 = heapq.heappop(ns)
        temp2 = heapq.heappop(ns)
        ans += temp1 + temp2
        heapq.heappush(ns, temp1 + temp2)
    print(ans)

32556KB

248ms

 

문제 출처 www.acmicpc.net/problem/1715

1715번: 카드 정렬하기