add: 拿出最少数目的魔法豆

Author: yi-ge

README.md

@@ -629,6 +629,10 @@ C++标准库提供的数据结构实在是太多了，参考[C++标准库头文
 
 ### 排序
 
+- [拿出最少数目的魔法豆](src/sort/removing_minimum_number_of_magic_beans.cpp)  [数组, 前缀和, 排序]
+
+  - LeetCode 2171. 拿出最少数目的魔法豆 <https://leetcode.cn/problems/removing-minimum-number-of-magic-beans>
+
 - [购买两块巧克力](src/sort/buy_two_chocolates.cpp)  [数组, 排序]
 
   - LeetCode 2706. 购买两块巧克力 <https://leetcode.cn/problems/buy-two-chocolates>

images/sort/removing_minimum_number_of_magic_beans.jpeg

@@ -0,0 +1,19 @@
+// 拿出最少数目的魔法豆
+// https://leetcode.cn/problems/removing-minimum-number-of-magic-beans
+// INLINE  ../../images/sort/removing_minimum_number_of_magic_beans.jpeg
+
+#include <headers.hpp>
+
+class Solution {
+public:
+  long long minimumRemoval(vector<int> &beans) {
+    int n = beans.size();
+    sort(beans.begin(), beans.end());
+    long long total = accumulate(beans.begin(), beans.end(), 0LL);
+    long long res = total;
+    for (int i = 0; i < n; i++) { // 遍历排序后的 beans 数组
+      res = min(res, total - (long long)beans[i] * (n - i));
+    }
+    return res;
+  }
+};

src/sort/removing_minimum_number_of_magic_beans.cpp

@@ -1,4 +1,4 @@
-// 执行编译时间：2024-01-17 13:55:48
+// 执行编译时间：2024-01-18 15:29:31
 #include <gtest/gtest.h>
 #include <lib.hpp>
 

test/lib/lib_test.cpp

@@ -0,0 +1,34 @@
+#include <sort/removing_minimum_number_of_magic_beans.cpp>
+
+TEST(拿出最少数目的魔法豆, minimumRemoval) {
+  Solution solution;
+  // 示例 1：
+  // 输入：beans = [4,1,6,5]
+  // 输出：4
+  // 解释：
+  // - 我们从有 1 个魔法豆的袋子中拿出 1 颗魔法豆。
+  //   剩下袋子中魔法豆的数目为：[4,0,6,5]
+  // - 然后我们从有 6 个魔法豆的袋子中拿出 2 个魔法豆。
+  //   剩下袋子中魔法豆的数目为：[4,0,4,5]
+  // - 然后我们从有 5 个魔法豆的袋子中拿出 1 个魔法豆。
+  //   剩下袋子中魔法豆的数目为：[4,0,4,4]
+  // 总共拿出了 1 + 2 + 1 = 4 个魔法豆，剩下非空袋子中魔法豆的数目相等。
+  // 没有比取出 4 个魔法豆更少的方案。
+  vector<int> beans = {4, 1, 6, 5};
+  EXPECT_EQ(solution.minimumRemoval(beans), 4);
+
+  // 示例 2：
+  // 输入：beans = [2,10,3,2]
+  // 输出：7
+  // 解释：
+  // - 我们从有 2 个魔法豆的其中一个袋子中拿出 2 个魔法豆。
+  //   剩下袋子中魔法豆的数目为：[0,10,3,2]
+  // - 然后我们从另一个有 2 个魔法豆的袋子中拿出 2 个魔法豆。
+  //   剩下袋子中魔法豆的数目为：[0,10,3,0]
+  // - 然后我们从有 3 个魔法豆的袋子中拿出 3 个魔法豆。
+  //   剩下袋子中魔法豆的数目为：[0,10,0,0]
+  // 总共拿出了 2 + 2 + 3 = 7 个魔法豆，剩下非空袋子中魔法豆的数目相等。
+  // 没有比取出 7 个魔法豆更少的方案。
+  beans = {2, 10, 3, 2};
+  EXPECT_EQ(solution.minimumRemoval(beans), 7);
+}