1. 解题思路
这一题只需要想清楚一个点就行了:
- 正常情况下,对于确定的n次选择,无论顺序如何,其得到的score都是相同的,要想要score出现差别,有需要有大量的应该减的score无法被减去,即那些孩子的happiness提前降至0了。
因此,我们的选取策略就是将孩子按照happiness进行排序,然后选择最大的k个孩子从大到小依次进行选取就行了,这样就能确保获得最大的score。
2. 代码实现
给出python代码实现如下:
class Solution:
def maximumHappinessSum(self, happiness: List[int], k: int) -> int:
happiness = sorted(happiness, reverse=True)
ans, dec = 0, 0
for h in happiness[:k]:
if h <= dec:
break
ans += h - dec
dec += 1
return ans
提交代码评测得到:耗时786ms,占用内存43.3MB。