add 2558

fishercoder1534 · fishercoder1534 · commit 32a2cb7277eb · 2023-02-05T13:18:42.000-08:00
diff --git a/README.md b/README.md
@@ -8,6 +8,7 @@ _If you like this project, please leave me a star._ &#9733;
 
 | #    |      Title     | Solutions                                                                                                                                | Video                                                                        | Difficulty                       | Tag                  
 |------|----------------|------------------------------------------------------------------------------------------------------------------------------------------|------------------------------------------------------------------------------|----------------------------------|-------------
+| 2558 |[Take Gifts From the Richest Pile](https://leetcode.com/problems/take-gifts-from-the-richest-pile/)| [Java](../master/src/main/java/com/fishercoder/solutions/_2558.java)                                                                     |                                          | Easy                             |
 | 2554 |[Maximum Number of Integers to Choose From a Range I](https://leetcode.com/problems/maximum-number-of-integers-to-choose-from-a-range-i/)| [Java](../master/src/main/java/com/fishercoder/solutions/_2554.java)                                                                     |                                          | Medium                           |
 | 2553 |[Separate the Digits in an Array](https://leetcode.com/problems/separate-the-digits-in-an-array/)| [Java](../master/src/main/java/com/fishercoder/solutions/_2553.java)                                                                     |                                          | Easy                             |
 | 2549 |[Count Distinct Numbers on Board](https://leetcode.com/problems/count-distinct-numbers-on-board/)| [Java](../master/src/main/java/com/fishercoder/solutions/_2549.java)                                                                     || Easy                                                                         |
diff --git a/src/main/java/com/fishercoder/solutions/_2558.java b/src/main/java/com/fishercoder/solutions/_2558.java
@@ -0,0 +1,23 @@
+package com.fishercoder.solutions;
+
+import java.util.PriorityQueue;
+
+public class _2558 {
+    public static class Solution1 {
+        public long pickGifts(int[] gifts, int k) {
+            PriorityQueue<Integer> maxHeap = new PriorityQueue<>((a, b) -> b - a);
+            for (int g : gifts) {
+                maxHeap.offer(g);
+            }
+            while (k-- > 0) {
+                int max = maxHeap.poll();
+                maxHeap.offer((int) Math.sqrt(max));
+            }
+            long res = 0l;
+            while (!maxHeap.isEmpty()) {
+                res += maxHeap.poll();
+            }
+            return res;
+        }
+    }
+}
