poj 2385 Apple Catching DP

標籤：dp

http://poj.org/problem?id=2385

狀態定義： dp[i][j] :=前i秒，移動j次接到的最大蘋果數量。

狀態轉移： dp[i][j]=dp[i-1][j]+num[0][i]; (j==0)

                   dp[i][j]=max{dp[i-1][j-1],dp[i-1][j]}+num[j%2==1][i]; (1=<j<=W)

這裡輸入有一個預先處理：開一個二維數組num[2][MAX_T]，num[0][i]：=第i秒樹1是否有蘋果下落，若有則為1，否則為0

num[1][i] :=第i秒樹2是否有蘋果下落，若有則為1，否則為0.

j%2==1判斷位置位於樹1還是2.

代碼：

#include<iostream>#include<cstring>#include<cstdlib>using namespace std;const int MAX_T=1001;const int MAX_W=31;int dp[MAX_T][MAX_W];int num[2][MAX_T];int main(){int T,W;while(cin>>T>>W){memset(num,0,sizeof(num));for(int i=1;i<=T;i++){int x;cin>>x;if(x==1) num[0][i]=1;else num[1][i]=1;}memset(dp,0,sizeof(dp));dp[1][0]=num[0][1];dp[1][1]=num[1][1];for(int i=2;i<=T;i++){for(int j=0;j<=W&&j<=i;j++){if(j==0) dp[i][j]=dp[i-1][j]+num[0][i];else dp[i][j]=max(dp[i-1][j-1],dp[i-1][j])+num[j%2==1][i];}}cout<<dp[T][W]<<endl;}return 0;} 

poj 2385 Apple Catching DP