动态编程:杆切割实现错误

时间:2012-05-02 18:51:21

标签: c++ algorithm dynamic-programming

给出一根长度为n英寸的杆和一张价格为pi的表格 i = 1,2,... n,确定通过切割可获得的最大收益 杆和销售件。

Bottom_Up_Cut_Rod(p, n)
1 let r[0...n] be a new array
2 r[0] = 0
3 for j = 1 to n
4 q = -infinity
5 for i = 1 to j
6 q = max(q; p[i] + r[j - i])
7 r[j] = q
8 return r[n]

实施

#include <iostream>
#include <algorithm>

using namespace std;

int RodCut(long long P[],long long n)
{
    long long r[n];
    r[0]=0;
    for(long long j=0;j<n;j++)
    {
         long long q = -100000;
         for(long long i=0;i<j;i++)
         {
             q = max(q , P[i] + r[j-i]);
         }
         r[j] = q;
    }

    return r[n];
}

int main()
{
    long long num;
    long long N;
    long long K;

    cin>>N;

    long long a[N];
    for (long long i = 0; i < N; i++)
    {
        cin>>num;
        a[i] = num;
    }

    int res = 0;
    res = RodCut(a,N);

    cout<<"Answer : "<<res;

    return 0;
}

我的输入为1 5 8 9 10 17 17 20 24 30,但输出为2686348。 我的代码出了什么问题?

2 个答案:

答案 0 :(得分:1)

有几个问题。您希望主循环从j = 1到n,因为它代表了使用j元素可以做到的最佳效果。

你应该坚持使用ints或long longs。

int r[n+1];
r[0]=0;

// Calculate best we can do with j elements
for(int j=1;j<=n;j++) {
    int q = -100000;
    for(int i=0;i<j;i++) {
        q = max(q , P[i] + r[j-i-1]);
    }
    r[j] = q;
}

return r[n];

这似乎为我提供了各种输入的正确解决方案。

答案 1 :(得分:0)

有两件事。一个是r[n],应该是r[n-1]。其次,开始j from 1 to n,因为r[0]在第一轮中被-100000取代。

此外,r[0]应为P[0];也就是说,如果长度为1的杆,你至少可以获得P[0]钱。

另请注意,q应为P[j],这是您所做的最低限度。

So assuming the array is P[0..n] // not including n
and r[0..n] is your memo for applying DP

foreach index from (0..n] // not including n
    r[index] = P[index]
    foreach splitIndex from (0..index] // not including index
        r[index] = max(r[index], P[splitIndex] + r[index-splitIndex-1]
return r[n-1]