Update maximal-score-after-applying-k-operations.cpp

Author: kamyu104

C++/maximal-score-after-applying-k-operations.cpp

@@ -9,6 +9,31 @@ class Solution {
             return (a + b - 1) / b;
         };
 
+        int64_t result = 0;
+        priority_queue<int> max_heap(cbegin(nums), cend(nums));
+        while (k-- && !empty(max_heap)) {
+            const auto x = max_heap.top(); max_heap.pop();
+            result += x;
+            const auto nx = ceil_divide(x, 3);
+            if (!nx) {
+                continue;
+            }
+            max_heap.emplace(nx);
+        }
+        return result;
+    }
+};
+
+// Time:  O(n + klogn)
+// Space: O(n)
+// heap
+class Solution2 {
+public:
+    long long maxKelements(vector<int>& nums, int k) {
+        const auto& ceil_divide = [](const auto& a, const auto& b) {
+            return (a + b - 1) / b;
+        };
+
         int64_t result = 0;
         priority_queue<int> max_heap(cbegin(nums), cend(nums));
         while (k--) {