题目大意是说有2棵苹果树,你可以选择站在其中的一棵树下,然后每秒在两棵树中的一棵可能有苹果落下,你每秒可以选择移动到另外一棵树或者不移动。只有当你在树下的时候才能接到苹果。现在题目给你共有k次移动的机会,然后问你最多可以得到多少个苹果。
其实算是比较简单的动态规划,转移方程式: dp[i][j][k]=max(dp[i-1][j][k]+1,dp[i-1][j^1][k-1]+1)(i表示第几秒,j表示哪棵树,k表示用了多少次移动。 如果当前树不为落下苹果的,那么删去+1
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=1100;
int t,w,a[maxn],dp[maxn][2][31];
int main()
{
while(scanf("%d%d",&t,&w)!=EOF)
{
for(int i=0;i<t;i++)
{
scanf("%d",&a[i]);
a[i]--;
}
for(int i=0;i<t;i++)
for(int k=0;k<=w;k++)
{
if(k==0)
{
dp[i][a[i]][k]=dp[i-1][a[i]][k]+1;
dp[i][a[i]^1][k]=dp[i-1][a[i]^1][k];
continue;
}
dp[i][a[i]][k]=max(dp[i-1][a[i]][k]+1,dp[i-1][a[i]^1][k-1]+1);
dp[i][a[i]^1][k]=max(dp[i-1][a[i]^1][k],dp[i-1][a[i]][k-1]);
}
int ans=0;
for(int i=0;i<=w;i++)
ans=max(ans,max(dp[t-1][0][i],dp[t-1][1][i]));
printf("%d\n",ans);
}
return 0;
}