LeetCode-3075 幸福值最大化的选择方案
Table of Contents
LeetCode-3075 幸福值最大化的选择方案 #
Solution 1 #
由于值下降到零就不再下降, 因此优先选值大的.
代码如下:
class Solution:
def maximumHappinessSum(self, happiness: List[int], k: int) -> int:
return sum([max(x - i, 0) for i, x in enumerate(sorted(happiness, key=lambda x: -x))][:k])