Question

问题描述：

给出一个整数列表和一个目标和，我们需要返回另一个包含布尔值的列表。这个布尔值列表表示我们正在寻找的子集。

示例：

输入：列表（3、34、4、12、5、2），目标=9。

输出：（false，false，true，false，true，false）（因为4 + 5 = 9；请注意，此解决方案不是唯一的，但是问题仅要求一个任意的解决方案）。

我的尝试

我首先编写了一种检查解决方案是否存在的方法：

boolean subsetSumRec(List<Integer> list, int i, int sum) { //i is the index of the last element in list
    if(sum==0){
        return true;
    } else if(i<0 || sum<0){
        return false;
    }
    boolean include = subsetSumRec(list,i-1,sum-list.get(n));
    boolean exclude = subsetSumRec(list,i-1,sum);
    return include || exclude;
}

我创建了一个新的ArrayList（类型：布尔型） resultList ，它将作为输出。我的计划是在方法subsetSumRec中更新 resultList 。就像这样：

private boolean subsetSumRec(List<Integer> list, int i, int sum) {
    if(sum==0){
        return true;
    } else if(list.size()<=0 || sum<0){
        return false;
    }
    res.add(true);
    boolean include = subsetSumRec(list,i-1,sum-list.get(i));
    boolean exclude;
    if(include == false){
        res.remove(true);
        res.add(false);
        exclude = subsetSumRec(list, i - 1, sum);
    }
    return include || exclude;
}

这没有给出所需的结果，但是我看不到出了什么问题。如何进行优化？

如果不是在subsetSumRec中更新 resultList 的选择，那么我就必须放弃检查是否存在解决方案并使用回溯算法的方法（无需动态编程！）。我不知道如何从最后一个开始。有人可以给我提示如何解决该问题的回溯算法（伪代码吗？）。

谢谢。

Answer 1

这是JavaScript中的一个示例递归。如果可行，我们将返回第二个列表，否则返回false。

function f(A, S, B=new Array(A.length).fill(false), i=A.length - 1){
  if (A[i] == S){
    B[i] = true
    return B
  }
    
  if (i == 0){
    if (A[i] == S){
      B[i] = true
      return B
    }
    return false
  }
  
  if (A[i] <= S){
    let maybeB = f(A, S - A[i], B, i - 1)
    if (maybeB){
      B[i] = true
      return B
    }
  }

  return f(A, S, B, i - 1)
}

var A = [3, 34, 4, 12, 5, 2]
var S = 9

console.log(JSON.stringify(A))
console.log(`Target: ${S}`)
console.log(JSON.stringify(f(A, S)))

S = 199
console.log(`\nTarget: ${S}`)
console.log(JSON.stringify(f(A, S)))

子集总和问题：返回所需子集的变体

1 个答案: