1. 程式人生 > >poj 1976 A Mini Locomotive

poj 1976 A Mini Locomotive

code clu ide -s || nbsp mat com lap

每個火車頭能拉走連續的k個貨箱,三個火車頭可以拉的最大貨箱重量是多少。

dp[i][j]表示i個火車頭在前j個火車頭裏可以拉的最大貨箱重量。

dp+記憶化搜索。

技術分享圖片
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
int _,n,k;
int dp[4][50007],a[50007],b[50007];//dp[i][j]表示前i個火車頭拉走前j個貨箱
int dfs(int hc,int ed){
    if
(hc<1||ed<1) return 0; if(dp[hc][ed]) return dp[hc][ed]; return dp[hc][ed]=max(dfs(hc,ed-1),dfs(hc-1,ed-k)+b[ed]);//從這個位置用一個火車頭拉走k個連續的貨箱或者不管這個位置 } int main(){ freopen("1.in","r",stdin); freopen("1.out","w",stdout); scanf("%d",&_); while(_--){ memset(dp,0,sizeof(dp)); scanf(
"%d",&n); for(int i=1;i<=n;i++) scanf("%d",&a[i]); for(int i=2;i<=n;i++) a[i]+=a[i-1];a[0]=0; scanf("%d",&k); for(int i=k;i<=n;i++) b[i]=a[i]-a[i-k]; printf("%d\n",dfs(3,n)); } return 0; }
View Code

poj 1976 A Mini Locomotive