这是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
请把正确的方向指向我做错了的地方。
答案 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];
}