子集和是我在 Macalester 学到的第一个 NP 完全问题。这个问题被查看了 36000 多次,但我没有看到足够的答案来详细解释算法的逻辑。所以我想我会尝试这样做。
假设:
为了简单起见,我首先假设输入集X 只包含正整数,而k 是正数。但是,我们可以调整算法来处理负整数以及k 是否为负的情况。
逻辑:
这个算法或真正的任何DP问题的关键是分解问题并简单地从基本情况开始。然后我们可以在基本情况上使用我们知道的一些知识:
- 我们知道,如果集合
X 为空,那么我们就无法求和k 的任何值。
- 如果一个集合
X 包含 k,那么它有一个子集和到 k。
- 我们知道,如果集合
x1 的一个子集是X 的子集,则与k1 相加,那么X 将有一个子集与k1 相加,即x1。
- 我们有一组
X = {x1, x1, x3, ......., xn, xn+1}。如果x1 = {x1, x1, x3, ......., xn} 有一个子集和到k - k1,我们知道它有一个子集和到k1。
举例说明1、2、3、4:
- 这很容易。如果您有一个空集 {}。因此你不能有一个子集
你不能有任何子集总和。
集合 X = {4} 的子集总和为 4,因为 4 它本身就是集合的一部分
假设您有一个集合x1 = {1,3,5},它是集合X = {1,3,5,2,8} 的一个子集。如果x1 与k1 = 8 有一个子集和,那么这意味着X 也有一个子集和为8,因为x1 是X 的子集
- 假设您有一个集合
X = {1,3,5,2,19},我们想知道它的子集总和是否为 20。它确实有,并且可以知道 x1 = {1,3,5,2} 是否为 (20 - 19) = 1 的一种方法。由于 x1 的子集总和为 1,因此当我们将 19 添加到集合 x1 时
我们可以用这个新数字 1 + 19 = 20 来创建我们想要的总和 20。
动态构建矩阵
凉爽的!现在让我们利用上述四个逻辑,从基本案例开始构建。我们将建立一个矩阵m。我们定义:
(请注意*的答案或大多数人构建函数 m(i,s) 但我认为矩阵是理解动态编程的一种简单方法。当我们在集合或数组中只有正数时它很有效. 但是函数路由更好,因为您不必处理超出范围的索引,匹配数组的索引并求和到矩阵......)
让我们用一个例子来构建矩阵:
X = {1,3,5,2,8}
k = 9
我们将逐行构建矩阵。我们最终想知道单元格 m[n][k] 包含true 或false。
第一行:
逻辑1.告诉我们矩阵的第一行应该都是false。
0 1 2 3 4 5 6 7 8 9
_ _ _ _ _ _ _ _ _ _
0| F F F F F F F F F F
1|
2|
3|
4|
5|
第二行及以上:
那么对于第二行或以上,我们可以使用逻辑 2,3,4 来帮助我们填充矩阵。
- 逻辑 2 告诉我们
m[i][s] = (X[i-1] == s)rememebr m[i] 指的是 X 中的第 i 个项目,即 X[i-1]
- 逻辑 3 告诉我们
m[i][s] = (m[i-1][s]) 这是在查看上面的单元格目录。
- 逻辑 4 告诉我们
m[i][s] = (m[i-1][s - X[i-1]]) 这是查看 X[i-1] 个单元格的上方和左侧的行。
如果其中任何一个是true,那么m[i][s] 就是true,否则false。所以我们可以将 2,3,4 改写成m[i][s] = (m[i-1][s] || a[i-1] == s || m[i-1][s - a[i-1]])
使用上述这些逻辑来填充矩阵m。在我们的示例中,它看起来像这样。
0 1 2 3 4 5 6 7 8 9
_ _ _ _ _ _ _ _ _ _
0| F F F F F F F F F F
1| F T F F F F F F F F
2| F T F T T F F F F F
3| F T F T T T T F T T
4| F T T T T T T T T T
5| F T T T T T T T T T
现在用矩阵来回答你的问题:
查看m[5][9],这是原始问题。使用前 5 个项目(即所有项目)我们可以找到 9 (k) 的子集总和吗?答案由true的单元格指示
代码如下:
import java.util.*;
public class SubSetSum {
public static boolean subSetSum(int[] a, int k){
if(a == null){
return false;
}
//n items in the list
int n = a.length;
//create matrix m
boolean[][] m = new boolean[n + 1][k + 1]; //n + 1 to include 0, k + 1 to include 0
//set first row of matrix to false. This also prevent array index out of bounds: -1
for(int s = 0; s <= k; s++){
m[0][s] = false;
}
//populate matrix m
for(int i = 1; i <= n; i++){
for(int s = 0; s <= k; s++){
if(s - a[i-1] >= 0){ //when it goes left we don't want it to go out of bounds. (logic 4)
m[i][s] = (m[i-1][s] || a[i-1] == s || m[i-1][s - a[i-1]]);
} else {
m[i][s] = (m[i-1][s] || a[i-1] == s);
}
}
}
//print matrix
print(m);
return m[n][k];
}
private static void print(boolean[][] m){
for(int i = 0; i < m.length; i++){
for(int j = 0; j < m[i].length; j++){
if(m[i][j]){
System.out.print("T");
} else {
System.out.print("F");
}
}
System.out.print("\n");
}
}
public static void main(String[] args){
int[] array = {1,3,5,2,8};
int k = 9;
System.out.println(subSetSum(array,k));
}
}
构建矩阵m 需要 O((n+1)(k+1)),即 O(nk)。看起来它应该是多项式的,但事实并非如此!它实际上是伪多项式。阅读它here
同样,这仅在输入仅包含正数时才有效。您可以轻松调整它以使用负数。矩阵仍然有 n+1 行,但有B - A + 1 列。其中B 是上限,A 是下限(+1 以包括零)。矩阵仍然是您必须将s 与下限偏移。
用文本从头到尾解释 DP 问题是相当困难的。但我希望这对那些试图理解这个问题的人有所帮助。
请注意,在上面的示例中,DP 表的行已排序。不必如此。
这里是问题案例的 DP 表,即给定一组 {5, 3, 11, 8, 2}。为简洁起见,我省略了错误值。
┌─────────┬──────┬──────┬──────┬──────┬──────┬──────┬──────┬──────┬──────┬──────┬──────┬──────┐
│ (index) │ 0 │ 2 │ 3 │ 5 │ 7 │ 8 │ 10 │ 11 │ 13 │ 14 │ 15 │ 16 │
├─────────┼──────┼──────┼──────┼──────┼──────┼──────┼──────┼──────┼──────┼──────┼──────┼──────┤
│ 0 │ true │ │ │ │ │ │ │ │ │ │ │ │
│ 5 │ true │ │ │ true │ │ │ │ │ │ │ │ │
│ 3 │ true │ │ true │ true │ │ true │ │ │ │ │ │ │
│ 11 │ true │ │ true │ true │ │ true │ │ true │ │ true │ │ true │
│ 8 │ true │ │ true │ true │ │ true │ │ true │ true │ true │ │ true │
│ 2 │ true │ true │ true │ true │ true │ true │ true │ true │ true │ true │ true │ true │
└─────────┴──────┴──────┴──────┴──────┴──────┴──────┴──────┴──────┴──────┴──────┴──────┴──────┘
下面是 JavaScript 中的一个实现,它将输出目标集 {5, 11}:
var subSetSum = function(input, sum) {
let y = input.length;
let x = sum;
if(input.length === 0) return 0;
let d = [];
//fill the rows
for (let i = 0; i <= y; i++) {
d[i] = [];
d[i][0] = true;
}
for (let j = 1; j <= y; j++) { //j row
for (let i = 1; i <= x; i++) { //i column
let num = input[j-1];
if(num === i) {
d[j][i] = true;
} else if(d[j-1][i]) {
d[j][i] = true;
} else if (d[j-1][i-num]) {
d[j][i] = true;
}
}
}
//console.table(d); //uncomment to see the table
if(!d[y][x]) return null;
let searchedSet = [];
for(let j=input.length, i=sum; j>0 && i != 0; j--) {
if(input[j-1] !== i) {
while(d[j-1][i]) { // go up
j--;
}
}
searchedSet.push(input[j-1]);
i = i-input[j-1];
}
return searchedSet;
};
console.log('searched set:'+ JSON.stringify(subSetSum([5, 3, 11, 8, 2], 16)));