POJ-1976(01背包)三个火车头

题目大意：有3个车头和一个车厢序列，每节车厢有一定的重量，每个车头最多可以拉连续的k节车厢，问3个车头最多可以拉的重量 

 解题思路：dp[i][j]表示第i节车厢，第j个车头能拉的最多的人数，由于车厢的重量不可能是负的因此，最优解一定满足每个车头要拉k节车厢 

代码如下

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
const int N=50005;
int dp[N][4],a[N];
int main()
{
    int T;
    cin>>T;
    while(T--){
        int n,k;
        a[0]=0;
        cin>>n;
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++){
            cin>>a[i];
            a[i]=a[i]+a[i-1];
        }
        cin>>k;
        /*for(int i=1;i<=n;i++){
            for(int j=1;j<=3;j++){
                int p=max(0,i-k);
                dp[i][j]=max(dp[i-1][j],dp[p][j-1]+a[i]-a[p]);
            }
        }*/
        for(int i=1;i<=n;i++){
            for(int j=1;j<=3;j++){
                if(i<j*k)
                    dp[i][j]=0;
                else
                    dp[i][j]=max(dp[i-1][j],dp[i-k][j-1]+a[i]-a[i-k]);
            }
        }
        cout<<dp[n][3]<<endl;
    }
    return 0;
}