poj 2385 Apple Catching 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;
}