Time: 1 ms (68.38%), Space: 13.1 MB (62.09%) - LeetHub

Shrutigupta03 · Shrutigupta03 · commit e81ef3ab53cb · 2024-12-13T00:12:27.000+05:30
diff --git a/2558-take-gifts-from-the-richest-pile/2558-take-gifts-from-the-richest-pile.cpp b/2558-take-gifts-from-the-richest-pile/2558-take-gifts-from-the-richest-pile.cpp
@@ -0,0 +1,23 @@
+class Solution {
+public:
+    long long pickGifts(vector<int>& gifts, int k) {
+        priority_queue<long long> pq;
+        
+        for(auto it : gifts) pq.push(it);
+        
+        while(k--){
+            long long temp = pq.top();
+            pq.pop();
+            temp = (long long)sqrt(temp);
+            pq.push(temp);
+        }
+        
+        long long res = 0;
+        
+        while(!pq.empty()){
+            res+=pq.top();
+            pq.pop();
+        }
+        return res;
+    }
+};
