Create minimum-cost-to-connect-sticks.py

Author: kamyu104

Python/minimum-cost-to-connect-sticks.py

@@ -0,0 +1,19 @@
+# Time:  O(nlogn)
+# Space: O(n)
+
+import heapq
+
+
+class Solution(object):
+    def connectSticks(self, sticks):
+        """
+        :type sticks: List[int]
+        :rtype: int
+        """
+        heapq.heapify(sticks)
+        result = 0
+        while len(sticks) > 1:
+            x, y = heapq.heappop(sticks), heapq.heappop(sticks)
+            result += x+y
+            heapq.heappush(sticks, x+y)
+        return result