题意:给出了一排车厢价值,现在有3个火车头,每个火车头都有限制拉的车厢数量,并且拉的车厢必须是连续的,问这三个火车头总共最大能拉多少?
题解:自己太菜了,没有想到,看了提示做的,也是最优化版01背包打多了,继续加油,回归正题,这个题该怎么做呢?无非就是每个连续子段选或者不选,那么背包的总容量为3跑01背包,dp[i][j]代表到了第i个车厢已经选了j个火车头的车厢总价值,递推方程为:
附上代码:
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=5e4+5;
int t,n,m,s[maxn];
int dp[maxn][5];
int sum[maxn];
int main()
{
scanf("%d",&t);
while(t--){
scanf("%d",&n);
memset(sum,0,sizeof(sum));
for(int i=1;i<=n;i++){
scanf("%d",&s[i]);
sum[i]=sum[i-1]+s[i];
}
scanf("%d",&m);
memset(dp,0,sizeof(dp));
for(int i=m;i<=n;i++){
for(int j=1;j<=3;j++){
dp[i][j]=max(dp[i-1][j],dp[i-m][j-1]+sum[i]-sum[i-m]);
}
}
printf("%d\n",dp[n][3]);
}
return 0;
}