-
Notifications
You must be signed in to change notification settings - Fork 254
/
Copy path06 Count of Subsets with given Sum.cpp
47 lines (40 loc) · 1.45 KB
/
06 Count of Subsets with given Sum.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
// https://www.geeksforgeeks.org/count-of-subsets-with-sum-equal-to-x/
#include <bits/stdc++.h>
using namespace std;
int CountSubsets(int arr[], int n, int sum) {
int t[n + 1][sum + 1]; // DP - matrix
// initialization
// here we are setting 1st row and 1st column
// i denotes the size of the array
// j denotes the target sum (subset sum)
for (int i = 0; i <= n; i++) {
for (int j = 0; j <= sum; j++) {
if (i == 0) // when array(i) is empty than there is no meaning of sum of elements so return count of subset as 0;
t[i][j] = 0;
if (j == 0) // when sum(j) is zero and there is always a chance of empty subset so return count as 1;
t[i][j] = 1;
}
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= sum; j++) {
if (arr[i - 1] <= j) // when element in the list is less then target sum
t[i][j] = t[i - 1][j - arr[i - 1]] + t[i - 1][j]; // either exclude or inxlude and add both of them to get final count
else
t[i][j] = t[i - 1][j]; // exclude when element in the list is greater then the sum
}
}
return t[n][sum]; // finally return the last row and last column element
}
signed main() {
int n; cin >> n;
int arr[n];
for (int i = 0; i < n; i++)
cin >> arr[i];
int sum; cin >> sum;
cout << CountSubsets(arr, n, sum) << endl;
return 0;
}
/*
Time Complexity: O(sum*n), where the sum is the ‘target sum’ and ‘n’ is the size of the array.
Auxiliary Space: O(sum*n), as the size of the 2-D array, is sum*n.
*/