题意
一个正整数数组 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;
}
};