poj 1976 A Mini Locomotive
阿新 • • 發佈:2018-09-11
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){ ifView Code(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; }
poj 1976 A Mini Locomotive