Create maximize-happiness-of-selected-children.cpp

Author: kamyu104

C++/maximize-happiness-of-selected-children.cpp

@@ -0,0 +1,18 @@
+// Time:  O(nlogn)
+// Space: O(1)
+
+// sort, greedy
+class Solution {
+public:
+    long long maximumHappinessSum(vector<int>& happiness, int k) {
+        sort(rbegin(happiness), rend(happiness));
+        int64_t result = 0;
+        for (int i = 0; i < k; ++i) {
+            if (happiness[i] - i <= 0) {
+                break;
+            }
+            result += happiness[i] - i;
+        }
+        return result;
+    }
+};