Skip to main content
  1. Posts/

LeetCode-3075 幸福值最大化的选择方案

·1 min·

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])