Merge pull request #735 from baekhangyeol/main

[백한결] 118차 라이브 코테 제출

Author: github-actions[bot]

live11/test118/문제1/백한결.py

@@ -0,0 +1,28 @@
+import sys
+import heapq
+
+def main():
+    input = sys.stdin.readline
+
+    T = int(input())
+
+    for _ in range(T):
+        K = int(input())
+        fileSize = list(map(int, input().split()))[:K]
+
+        heapq.heapify(fileSize)
+        total = 0
+
+        while len(fileSize) >= 2:
+            first = heapq.heappop(fileSize)
+            second = heapq.heappop(fileSize)
+            cost = first + second
+            total += cost
+
+            heapq.heappush(fileSize, cost)
+
+        print(total)
+
+
+if __name__ == '__main__':
+    main()

live11/test118/문제2/백한결.py

@@ -0,0 +1,28 @@
+import sys
+import heapq
+
+def main():
+    input = sys.stdin.readline
+
+    N = int(input())
+    cardSize = []
+
+    for _ in range(N):
+        card = int(input())
+        cardSize.append(card)
+
+    heapq.heapify(cardSize)
+    total = 0
+
+    while len(cardSize) >= 2:
+        first = heapq.heappop(cardSize)
+        second = heapq.heappop(cardSize)
+        cost = first + second
+        total += cost
+
+        heapq.heappush(cardSize, cost)
+
+    print(total)
+
+if __name__ == '__main__':
+    main()