POJ 1976 A Mini Locomotive
題意:給出n車廂以及n節車廂內的人數 現有三節迷你火車頭 每個火車頭可以拉連續的m節車廂 問三個火車頭最多可以拉多少人。
dp[i][j]表示第i節車廂有j個車頭可以拉的總人數 dp[i][j]=max(dp[i-1][j],dp[k][j-1]+a[i]-a[k]) k=max(i-m,0) 前一個表示該節車廂不拉 後一狀態表示加一個火車頭拉
#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int a[50005],dp[50005][4];
int main()
{
int t,n,m,k;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
memset(dp,0,sizeof(dp));
a[0]=0;
for(int i=1; i<=n; i++)
scanf("%d",&a[i]),a[i]+=a[i-1];
scanf("%d",&m);
for(int i=1; i<=n; i++)
for(int j=1; j<4; j++)
{
k=i-m<0?0:i-m;
dp[i][j]=max(dp[i-1][j],dp[k][j-1]+a[i]-a[k]);
}
printf("%d\n",dp[n][3]);
}
return 0;
}
最後更新:2017-04-03 12:53:49