每日一题 2171. 拿出最少数目的魔法豆（中等）

最新推荐文章于 2024-10-02 23:40:03 发布

eyvr

最新推荐文章于 2024-10-02 23:40:03 发布

阅读量411

点赞数 9

分类专栏：用Python刷力扣文章标签：算法 leetcode python

本文链接：https://blog.csdn.net/qq_46636391/article/details/135681469

版权

用Python刷力扣专栏收录该内容

108 篇文章 1 订阅

订阅专栏

该篇文章介绍了一种解决关于魔法豆分配问题的算法，通过将豆子数量按升序排列，计算每次取走当前最少数量后剩余豆子所需的额外豆子，最后返回最小操作次数以使剩余豆子袋数等于最少豆子数量。

摘要由CSDN通过智能技术生成

在这里插入图片描述

只有两种拿法，要么全拿光，剩下的要取到和最少魔法豆的袋子的数目相等，显然关键在于每个袋子的最少豆子数量是多少（少于这个数量的袋子是全拿出的）
想到了从小到大排序先把最少豆子的袋子认为是基线（最少数量），计算要拿几颗，然后再以下一个排好序的袋子为基线，再算要拿几颗，遍历完数组就可以得到答案了

class Solution:
    def minimumRemoval(self, beans: List[int]) -> int:
        beans.sort()
        n = len(beans)
        ans = [0] * n
        ans[0] = sum(beans) - n * beans[0]
        for i in range(1, n):
            ans[i] = ans[i - 1] + beans[i - 1] - (beans[i] - beans[i - 1]) * (n - i)
        return min(ans)