POJ2385 简单DP

题意 两棵苹果树，每一分钟某一颗树掉一个苹果，一个人接苹果，她每一次掉落苹果时只能在一棵树下等待。给你总时间T，人最多可以在两棵树间移动的次数W，还有每分钟是哪棵树掉苹果，问你她最多接多少个苹果。

思路 dp(i,j,k)表示在前i分钟，最多移动j次，并且在i分钟站在第k棵下时最多接到的苹果数。a[i]表示i分钟时哪棵树掉。

   状态转移 dp(i,j,a[i]) = max(dp(i-1,j,a[i]) ,dp(i-1,j-1,a[i]^1)) + 1;  

                          dp(i,j,a[i]^1) = max(dp(i-1,j-1,a[i]) ,dp(i-1,j,a[i]^1));

   利用滚动数组思想，在实现时可以去掉i这一维。

   

我这个算法应该还不是最好的，不过对这题足够了，以后有时间可以再优化下~

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
#define MAXN 1005
#define MAXM 35

int dp[MAXM][2];
int n,m;
int a[MAXN];


int main()
{
	int i,j;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		a[i]--; 
	} 
	for(i=1;i<=n;i++)
	{
		dp[0][a[i]] = dp[0][a[i]] + 1;
		dp[0][a[i]^1] = dp[0][a[i]^1];
		for(j=1;j<=m;j++)
		{
			dp[j][a[i]] = max(dp[j][a[i]],dp[j-1][a[i]^1]) + 1;
			dp[j][a[i]^1] = max(dp[j-1][a[i]],dp[j][a[i]^1]);
		}
	}
	cout<<max(dp[m][0],dp[m][1])<<endl;
	return 0;
}