POJ_2385_Apple Catching_简单DP

听了一天古典钢琴曲真不错

题意：

农夫约翰又养了一头极吊的牛Bessie，他喜欢吃苹果，有两棵苹果树，从1到T秒，每分钟其中一棵树会掉苹果，Bessie每分钟站在其中一棵树下，如果此时掉苹果，就能吃到，他最多移动W次，移动不算时间，问T分钟后他最多吃到多少个苹果。

Input
* Line 1: Two space separated integers: T(1000） and W（30）

* Lines 2..T+1: 1 or 2: the tree that will drop an apple each minute.
Output
* Line 1: The maximum number of apples Bessie can catch without walking more than W times.

如果用搜索，那么至少有2^w个状态，肯定TLE，动态规划，记录当前时间、当前位置、当前移动次数即可

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
using namespace std;
#define mxt 1010
#define mxw 40
int t,w;
int a[mxt];
int dp[2][mxt][mxw];
int main(){
scanf("%d%d",&t,&w);
for(int i=1;i<=t;++i)	scanf("%d",&a[i]);
memset(dp,0,sizeof(dp));
int ans=0;
for(int j=1;j<=t;++j)
for(int i=0;i<2;++i)
for(int k=0;k<=min(j-1,w);++k){
dp[i][j][k]=dp[i][j-1][k]+(a[j]==i+1 ? 1 : 0);
if(k-1>=0)
dp[i][j][k]=max(dp[i][j][k],dp[1-i][j-1][k-1]+(a[j]==i+1 ? 1 : 0));
ans=max(ans,dp[i][j][k]);
}
printf("%d\n",ans);
return 0;
}