题目大意:有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;
}