740. Delete and earn

740-delete-and-earn.py

import collections

class Solution:
    def deleteAndEarn(self, nums: List[int]) -> int:
        m: int = max(nums)
        buckets: List[int] = [0 for _ in range(m+1)]
        counter = collections.Counter(nums)
        for k, v in counter.items():
            buckets[k] = v*k
        take, skip = 0, 0    
        for i in range(m+1):
            take_i = skip + buckets[i]
            skip_i = max(skip, take)
            skip = skip_i
            take = take_i
        return max(take, skip)