我正在研究经典的DP问题,计数子集总和。
如果不确定该问题,则问题陈述如下:给定一组正数,找到总和等于给定数“ S”的子集总数。
我正在尝试使用回溯方法来解决它。
假设给定数组为[1, 1, 2, 3]
,sum
为4,则预期结果为3
奇怪的是,代码打击效果很好,并且可以生成正确的结果。
private int backtrack2(int[] nums, int sum) {
int[] count = new int[1];
backtrack2Helper(nums, sum, count, 0);
return count[0];
}
private int backtrack2Helper(int[] nums, int sum, int[] count, int pos) {
if (0 == sum) return 1;
if (pos == nums.length) return 0;
for (int i = pos; i < nums.length; ++i) {
if (sum - nums[i] >= 0) {
int temp = backtrack2Helper(nums, sum - nums[i], count, i + 1);
count[0] += temp;
}
}
return 0;
}
但是,下面的代码无法生成正确的结果:
private int backtrack2(int[] nums, int sum) {
int[] count = new int[1];
backtrack2Helper(nums, sum, count, 0);
return count[0];
}
private int backtrack2Helper(int[] nums, int sum, int[] count, int pos) {
if (0 == sum) return 1;
if (pos == nums.length) return 0;
for (int i = pos; i < nums.length; ++i) {
if (sum - nums[i] >= 0) {
count[0] += backtrack2Helper(nums, sum - nums[i], count, i + 1);
}
}
return 0;
}
唯一的区别是辅助函数中的if
语句部分。
int temp = backtrack2Helper(nums, sum - nums[i], count, i + 1);
count[0] += temp;
和
count[0] += backtrack2Helper(nums, sum - nums[i], count, i + 1);
应该等效,对吧?
但是第二个不能正常工作。我真的很困惑,我一步一步地调试了它,发现当第二种方法中的方法最后到达return语句时,返回的0并没有添加count[0]
,而是设置为0。
发生了什么事?
差异是由于backtrack2Helper()
更改了count[0]
。
例如,假设backtrack2Helper()将count [0]的值从0更改为1并返回1。
在一个摘要中:
int temp = backtrack2Helper(nums, sum - nums[i], count, i + 1); // == 1
count[0] += temp; // == 1 + 1 == 2
但另一个:
count[0] += backtrack2Helper(nums, sum - nums[i], count, i + 1); // == 0 + 1 == 1