动态规划的经典题目
注意: 人开始的时候在tree1下面,这样如果给出显示第2颗树落下苹果的话,初始化代码就得改变,所以分两种情况。
状态数组dp[i][j]其中的i表示时间,W表示的次数;
dp状态方程dp[i][j] = max(dp[i-1][j], dp[i-1][j-1]);
这样的话苹果数目是不变的,假如某时刻到达一棵树下且这树掉下了苹果的话就dp[i][j]++;
参考代码如下:
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <math.h>
#include <algorithm>
using namespace std;
int arr[1005];
int dp[1005][35];
int main()
{
int T,W;
while(scanf("%d%d",&T,&W)!=EOF)
{
memset(dp,0,sizeof(dp));
for(int i=1;i<=T;i++)
scanf("%d",&arr[i]);
if(arr[1]==1)
{
dp[1][0]=1;
dp[1][1]=0;
}
if(arr[1]==2)
{
dp[1][1]=1;
dp[1][0]=0;
}
for(int i=2;i<=T;i++)
for(int j=0;j<=W;j++)
{
if (j == 0)
{
dp[i][j] = dp[i - 1][j] + arr[i] % 2;
continue;
}
dp[i][j] = max(dp[i - 1][j] ,dp[i - 1][j - 1]) ;
if(j%2+1==arr[i])
dp[i][j]++;
}
int ans = dp[T][0];
for (int i = 0; i <= W ; ++ i)
{
if (ans < dp[T][i])
{
ans = dp[T][i];
}
}
printf("%d\n", ans);
}
}