POJ 2385 Apple Catching DP

用opt[i][j]表示第i分钟走了j次后最多能摘几个苹果

由于只有两棵树，故用j%2+1就能判断此时Bessie在哪棵树下面，省下了一维空间

f[i]表示第i分钟哪颗树掉苹果，若f[i]==j%2+1,则此时可以摘得苹果，opt[i][j]=max(opt[i-1][j],opt[i-1][j-1])+1

否则，此时摘不到苹果，opt[i][j]==max(opt[i-1][j],opt[i-1][j-1])

但对于opt[i][0]要单独考虑，即第i分钟时仍然没移动，在第一棵树下的情况，可由opt[i-1][0]推导

 

代码：

#include<iostream> using namespace std; int opt[1001][31]; int f[1001]; int max(int a,int b) { return a>b?a:b; } int main() { int t,w,i,j,k; cin>>t>>w; for(i=1;i<=t;i++) cin>>f[i]; for(i=1;i<=t;i++) { opt[i][0]=opt[i-1][0]; if(f[i]==1) opt[i][0]++; for(j=1;j<=w&&j<=i;j++) { if(j%2+1==f[i])//能摘到苹果 { opt[i][j]=max(opt[i-1][j-1]+1,opt[i-1][j]+1); } else//不能摘到苹果 { opt[i][j]=max(opt[i-1][j-1],opt[i-1][j]); } } } k=opt[t][1]; for(i=2;i<=w;i++) if(opt[t][i]>k) k=opt[t][i]; cout<<k<<endl; return 0; }