LC-2902 Hard LeetCode

2902. Count of Sub-Multisets With Bounded Sum

Array Hash Table Dynamic Programming Sliding Window

Read the full problem statement on LeetCode.

Difficulty: hard Acceptance: 20% Topics: Array, Hash Table, Dynamic Programming, Sliding Window

Reading material

Arrays & Hashing 12 min 1-D Dynamic Programming 16 min Sliding Window 11 min

Reference solution (spoiler · python)

# Time:  O(n + d * r), d = len(set(nums))
# Space: O(d + r)

import collections


# freq table, knapsack dp, sliding window, combinatorics
class Solution(object):
    def countSubMultisets(self, nums, l, r):
        """
        :type nums: List[int]
        :type l: int
        :type r: int
        :rtype: int
        """
        MOD = 10**9+7
        cnt = collections.Counter(nums)
        dp = [0]*(r+1)
        dp[0] = 1
        for x, c in cnt.iteritems():
            for i in reversed(xrange(max(r-x+1, 1), r+1)):
                curr = reduce(lambda x, y: (x+y)%MOD, (dp[i-x*j] for j in xrange(min(c, i//x+1))))
                for j in reversed(xrange((i-1)%x+1, i+1, x)):
                    curr = (curr+(dp[j-x*c] if j-x*c >= 0 else 0)-dp[j])%MOD
                    dp[j] = (dp[j]+curr)%MOD
        return (reduce(lambda x, y: (x+y)%MOD, (dp[i] for i in range(l, r+1)))*(cnt[0]+1))%MOD

Solution from kamyu104/LeetCode-Solutions · MIT