子集和问题的回溯剪枝解法
1. 问题
1.1 原文描述
问题参考的是 程振波老师的 《Design and Analysis of Algorithm Using Python》2.5.3节的描述:
输入给定整数集,问是否存在该整数集的子集,使得该子集元素的和为 0。
比如给定的整数集合为 [-7, -3 ,-2, 5, 8],存在集合的子集 [-3, -2, 5],该子集和为0。
1.2 IO示例
input:
-2 -3 4 5
0
output:
1101
0000
2. 解法
2.1 传统解法
这里不给出代码,如果有兴趣可以去看一下 程老师 的书。
传统解法的思路是:直接穷举出所有子集合,保留其和为目标数的子集合。
这种做法无论空间复杂度还是时间复杂度都是 O ( 2 n ) O(2^n) O(2n),可谓效率低下,事倍功半。
2.2 优化
这里给出优化的解法,
遗憾的是,这种解法并不能降低算法的时间复杂度
这只能在原本的复杂度下做出优化,减少在某些情况下的计算次数。
这种解法被称作回溯剪枝
2.3 算法描述
规定: x i x_i xi为第 i i i 层的栈信息(0 or 1)
w i w_i wi为第 i i i 层的数字信息(即输入数组第 i i i 个元素的大小)
t t t 是目标数
向右遍历 0 不选
向左遍历 1 选
-
初始化 栈(用来保存轨迹)、中间结果。
-
递归遍历状态树 规定:目前的深度为 k k k
- 如果中间结果与目标结果一致,保存栈信息
- 若 ∑