Add Add-all solution

thorvn · Oct 7, 2022 · e379b5b · e379b5b
1 parent 5ec8b69
commit e379b5b
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/Lecture-07-heap/add-all.py b/Lecture-07-heap/add-all.py
@@ -0,0 +1,17 @@
+import heapq
+
+n = int(input())
+
+while ( n!=0 ):
+  h = list(map(int, input().split()))
+  heapq.heapify(h)
+
+  cost = 0
+  while len(h) > 1:
+    sum = heapq.heappop(h) + heapq.heappop(h)
+    heapq.heappush(h, sum)
+    cost += sum
+
+  print(cost)
+
+  n = int(input())