题目大意:有两棵苹果树,每秒都会有一颗苹果树掉苹果下来,人一开始在第一颗树下,只有w次移动的机会,求最多能拿几颗苹果。
解题思路:dp[i][j]代表在第i课树下移动了j次获得的最多果子,由于一开始在第一颗树下,所以初始化要注意一下。
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int inf=0x3f3f3f3f;
int t,w,a[1005];
int dp[5][1005];
int main() {
while(cin>>t>>w) {
for(int i=0; i<t; i++) {
cin>>a[i];
}
memset(dp,0,sizeof dp);
int ans=0;
if(a[0]==1) dp[1][0]=1;
else dp[2][1]=1;
for(int i=1; i<t; i++) {
for(int j=0; j<=w; ++j) {
if(j){
dp[a[i]][j]=max(dp[a[i]][j],dp[a[i]==1?2:1][j-1])+1;
}
else{
dp[a[i]][j]=dp[a[i]][j]+1;
}
ans=max(ans,dp[a[i]][j]);
}
}
cout<<ans<<endl;
}
return 0;
}