6006. 拿出最少数目的魔法豆

题意

一个正整数数组 beans， 每位都有 beans[i] 个豆。从每位拿出一些豆，使得非0位的值都相同，求最少拿出的豆数目。

样例

输入：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 个魔法豆更少的方案。

输入：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 从小到大排序后，枚举最终所有非空位豆的数目 x，小于 x 的豆清空，大于 x 的豆减少到 x。

结果：0000xxxxx, 需要拿掉的即为：ans = sum - (len - i) * x;

算法

思维 + 排序

代码

class Solution {
public:
    long long minimumRemoval(vector<int>& beans) {
        int len = beans.size();
        long long sum = 0, res;
        for (auto &x: beans) sum += x;
        res = sum;
        sort(beans.begin(), beans.end());
        for (int i = 0; i < len; i++) {
            res = min(res, sum - (len - i) * (long long)beans[i]);
        }
        return res;
    }
};