Create 1167.Minimum-Cost-to-Connect-Sticks.cpp

wisdompeak · web-flow · commit a1044cb0f767 · 2021-05-29T01:28:35.000-07:00
diff --git a/Heap/1167.Minimum-Cost-to-Connect-Sticks/1167.Minimum-Cost-to-Connect-Sticks.cpp b/Heap/1167.Minimum-Cost-to-Connect-Sticks/1167.Minimum-Cost-to-Connect-Sticks.cpp
@@ -0,0 +1,22 @@
+class Solution {
+public:
+    int connectSticks(vector<int>& sticks) 
+    {
+        multiset<int>Set;
+        int sum = 0;
+        for (auto x: sticks)
+        {
+            Set.insert(x);
+        }
+        for (int i=0; i<sticks.size()-1; i++)
+        {
+            int x = *(Set.begin());
+            Set.erase(Set.begin());
+            int y = *(Set.begin());
+            Set.erase(Set.begin());
+            sum += x+y;
+            Set.insert(x+y);
+        }
+        return sum;
+    }
+};
