让is_subset_sum(int set[], int n, int sum) 成为函数来查找是否存在set[] 的子集且总和等于总和。 n 是set[] 中的元素个数。
is_subset_sum problem 可以分为两个子问题
- 包括最后一个元素,recur for n = n-1, sum = sum – set[n-1]
- 排除最后一个元素,在 n = n-1 时重复。
如果上述任何子问题返回 true,则返回 true。
以下是 is_subset_sum() 问题的递归公式。
is_subset_sum(set, n, sum) = is_subset_sum(set, n-1, sum) || is_subset_sum(set, n-1, sum-set[n-1])
Base Cases:
is_subset_sum(set, n, sum) = false, if sum > 0 and n == 0
is_subset_sum(set, n, sum) = true, if sum == 0
我们可以使用动态规划解决Pseudo-polynomial time 中的问题。我们创建一个布尔二维表子集[][] 并以自下而上的方式填充它。如果 set[0..j-1] 的一个子集的和等于 i.,则子集 [i][j] 的值将为真,否则为假。最后,我们返回子集[sum][n]
解的时间复杂度是O(sum*n)。
用 C 实现
// A Dynamic Programming solution for subset sum problem
#include <stdio.h>
// Returns true if there is a subset of set[] with sun equal to given sum
bool is_subset_sum(int set[], int n, int sum) {
// The value of subset[i][j] will be true if there is a
// subset of set[0..j-1] with sum equal to i
bool subset[sum+1][n+1];
// If sum is 0, then answer is true
for (int i = 0; i <= n; i++)
subset[0][i] = true;
// If sum is not 0 and set is empty, then answer is false
for (int i = 1; i <= sum; i++)
subset[i][0] = false;
// Fill the subset table in botton up manner
for (int i = 1; i <= sum; i++) {
for (int j = 1; j <= n; j++) {
subset[i][j] = subset[i][j-1];
if (i >= set[j-1])
subset[i][j] = subset[i][j] || subset[i - set[j-1]][j-1];
}
}
/* // uncomment this code to print table
for (int i = 0; i <= sum; i++) {
for (int j = 0; j <= n; j++)
printf ("%4d", subset[i][j]);
printf("\n");
} */
return subset[sum][n];
}
// Driver program to test above function
int main() {
int set[] = {3, 34, 4, 12, 5, 2};
int sum = 9;
int n = sizeof(set)/sizeof(set[0]);
if (is_subset_sum(set, n, sum) == true)
printf("Found a subset with given sum");
else
printf("No subset with given sum");
return 0;
}