3082. Find the Sum of the Power of All Subsequences
Asked by 2 companies
Topics
You are given an integer array nums of length n and a positive integer k.
The power of an array of integers is defined as the number of subsequences with their sum equal to k.
Return the sum of power of all subsequences of nums.
Since the answer may be very large, return it modulo 109 + 7.
Example 1:
Input: nums = [1,2,3], k = 3
Output: 6
Explanation:
There are 5 subsequences of nums with non-zero power:
- The subsequence
[1,2,3]has2subsequences withsum == 3:[1,2,3]and[1,2,3]. - The subsequence
[1,2,3]has1subsequence withsum == 3:[1,2,3]. - The subsequence
[1,2,3]has1subsequence withsum == 3:[1,2,3]. - The subsequence
[1,2,3]has1subsequence withsum == 3:[1,2,3]. - The subsequence
[1,2,3]has1subsequence withsum == 3:[1,2,3].
Hence the answer is 2 + 1 + 1 + 1 + 1 = 6.
Example 2:
Input: nums = [2,3,3], k = 5
Output: 4
Explanation:
There are 3 subsequences of nums with non-zero power:
- The subsequence
[2,3,3]has 2 subsequences withsum == 5:[2,3,3]and[2,3,3]. - The subsequence
[2,3,3]has 1 subsequence withsum == 5:[2,3,3]. - The subsequence
[2,3,3]has 1 subsequence withsum == 5:[2,3,3].
Hence the answer is 2 + 1 + 1 = 4.
Example 3:
Input: nums = [1,2,3], k = 7
Output: 0
Explanation: There exists no subsequence with sum 7. Hence all subsequences of nums have power = 0.
Constraints:
1 <= n <= 1001 <= nums[i] <= 1041 <= k <= 100
Hints
Hint 1
j with the sum of elements k, it contributes 2n - j to the answer.Hint 2
dp[i][j] represent the number of subsequences in the subarray nums[0..i] which have a sum of j.Hint 3
dp[i][k] for all 0 <= i <= n-1 and multiply them with 2n - j to get final answer.