hdu 1244 Max Sum Plus Plus Plus

http://acm.hdu.edu.cn/showproblem.php?pid=1244

状态转移方程：dp[i][j]=max(dp[i][j-1],dp[i-1][j-a[i]]+sum[j]-sum[j-a[i]]);

dp[i][j]为第i段第j个数。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 2000
using namespace std;

int dp[30][maxn];
int a[maxn],b[maxn],sum[maxn];
int n,m;

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0) break;
        scanf("%d",&m);
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        memset(dp,0,sizeof(dp));
        memset(sum,0,sizeof(sum));
        for(int i=1; i<=m; i++)
        {
            scanf("%d",&a[i]);
        }
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&b[i]);
            sum[i]=sum[i-1]+b[i];
        }
        int len=0;
        for(int i=1; i<=m; i++)
        {
            len+=a[i];
            for(int j=len; j<=n; j++)
            {
                dp[i][j]=max(dp[i][j-1],dp[i-1][j-a[i]]+sum[j]-sum[j-a[i]]);
            }
        }
        printf("%d
",dp[m][n]);
    }
    return 0;
}