题目链接:http://poj.org/problem?id=2385
状态定义: dp[i][j] :=前i秒,移动j次接到的最大苹果数量。
状态转移: dp[i][j]=dp[i-1][j]+num[0][i]; (j==0)
dp[i][j]=max{dp[i-1][j-1],dp[i-1][j]}+num[j%2==1][i]; (1=<j<=W)
这里输入有一个预处理:开一个二维数组num[2][MAX_T],num[0][i]:=第i秒树1是否有苹果下落,若有则为1,否则为0
num[1][i] :=第i秒树2是否有苹果下落,若有则为1,否则为0.
j%2==1判断位置位于树1还是2.
代码:
#include<iostream>
#include<cstring>
#include<cstdlib>
using namespace std;
const int MAX_T=1001;
const int MAX_W=31;
int dp[MAX_T][MAX_W];
int num[2][MAX_T];
int main()
{
int T,W;
while(cin>>T>>W)
{
memset(num,0,sizeof(num));
for(int i=1;i<=T;i++){
int x;
cin>>x;
if(x==1) num[0][i]=1;
else num[1][i]=1;
}
memset(dp,0,sizeof(dp));
dp[1][0]=num[0][1];
dp[1][1]=num[1][1];
for(int i=2;i<=T;i++){
for(int j=0;j<=W&&j<=i;j++){
if(j==0) dp[i][j]=dp[i-1][j]+num[0][i];
else
dp[i][j]=max(dp[i-1][j-1],dp[i-1][j])+num[j%2==1][i];
}
}
cout<<dp[T][W]<<endl;
}
return 0;
}