使用DP

时间:2019-06-12 06:01:00

标签: c++ bottom-up

这是DP的常见问题。我必须在仅由非相邻元素组成的数组中找到最大和。我看到了其他文章,但我想使用动态编程来做到这一点,尤其是自下而上的DP。这是我的代码:

int maxSubsetSum(vector<int> arr,int start,int end) {
    int sum[arr.size()]; //to store max sum upto a given index
    sum[0] = arr[start];
    sum[1] = max(arr[start], arr[end]);
    if(start==end){
      return sum[0];
    }
    else if(start==end-1){
      return sum[1];
    }
    else{
        for(int i=2;i<=end;i++){
            sum[i]=max(arr[i],max(arr[i]+sum[i-2],sum[i-1]));
        }
    }
    return sum[end];
}

我在通过测试用例时遇到错误。尽管当我手动运行一个测试用例时,它还是可以正常工作的。实际输出与我获得的输出相同。但是测试系统为我的代码提供了不同的输出。

我在测试用例上手动测试了此代码:3 5 -7 8 10 并且答案与实际输出(= 15)匹配,但测试用例没有通过。

sum[0]=3
sum[1]=5
sum[2]=max(-7,max(-7+3,5))=5
sum[3]=max(8,max(8+5,5))=13
sum[4]=max(10,max(10+5,13))=15

请把正确的方向指向我做错了的地方。

1 个答案:

答案 0 :(得分:3)

尝试一下:

int maxSubsetSum(vector<int> arr) {
    if (arr.size() == 1) {
      return arr[0];
    }
    if (arr.size() == 2) {
      return max(arr[0], arr[1]);
    }
    int sum[arr.size()]; //to store max sum upto a given index
    sum[0] = arr[0];
    sum[1] = max(arr[0], arr[1]);
    for(int i = 2; i < arr.size(); i++) {
        sum[i] = max(arr[i], max(arr[i] + sum[i - 2], sum[i - 1]));
    }
    return sum[arr.size() - 1];
}