3075. Maximize Happiness of Selected Chi

[흑화뉴비(coderhs)]

class Solution:

    def maximumHappinessSum(self, happiness: List[int], k: int) -> int:

        return sum([max(x - i, 0) for i, x in enumerate(sorted(happiness, reverse=True)[:k])])

오랜만에 1-Liner