子集和是我在 Macalester 学到的第一个 NP 完全问题。这个问题被查看了 36000 多次,但我没有看到足够的答案来详细解释算法的逻辑。所以我想我会尝试这样做。
假设:
为了简单起见,我首先假设输入集X
只包含正整数并且k
是正数。但是,我们可以调整算法以处理负整数以及 ifk
为负数的情况。
逻辑:
该算法或任何 DP 问题的关键是分解问题并简单地从基本情况开始。然后我们可以使用我们知道的一些知识在基本案例的基础上进行构建:
- 我们知道,如果集合
X
是空的,那么我们就无法求和 的任何值k
。
- 如果一个集合
X
包含k
那么它有一个子集总和k
。
- 我们知道,如果集合的
x1
一个子集是X
sum to k1
then的子集,那么它的子集就是X
sum 。k1
x1
- 我们有一套
X = {x1, x1, x3, ......., xn, xn+1}
。我们知道它有一个子集和到k1
如果x1 = {x1, x1, x3, ......., xn}
有一个子集和到k - k1
。
举例说明 1,2,3,4:
- 这很容易。如果您有一个空集 {}。你不能有一个子集,因此你不能有任何子集总和。
一个集合X = {4}
的子集总和为 4,因为 4 它本身就是集合的一部分
假设您有一个 setx1 = {1,3,5}
是 set 的子集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。
动态构建矩阵
Cool!现在让我们利用上述四个逻辑,从基本案例开始构建。我们将建立一个矩阵m
。我们定义:
(注意维基百科的答案或大多数人构建了一个函数 m(i,s) 但我认为矩阵是理解动态编程的一种简单方法。当我们在集合或数组中只有正数时它工作得很好。但是函数路由更好,因为您不必处理超出范围的索引,匹配数组的索引并求和到矩阵......)
让我们使用示例构建矩阵:
X = {1,3,5,2,8}
k = 9
我们将逐行构建矩阵。我们最终想知道单元 m[n][k] 包含true
or 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)。看起来它应该是多项式的,但事实并非如此!它实际上是伪多项式。在这里阅读
同样,这仅在输入仅包含正数时才有效。您可以轻松调整它以使用负数。矩阵仍然有 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)));