[洛谷]P2690 接苹果 (#线性dp)

题目描述

很少有人知道奶牛爱吃苹果。农夫约翰的农场上有两棵苹果树（编号为1和2）， 每一棵树上都长满了苹果。奶牛贝茜无法摘下树上的苹果，所以她只能等待苹果 从树上落下。但是，由于苹果掉到地上会摔烂，贝茜必须在半空中接住苹果（没有人爱吃摔烂的苹果）。贝茜吃东西很快，她接到苹果后仅用几秒钟就能吃完。每一分钟，两棵苹果树其中的一棵会掉落一个苹果。贝茜已经过了足够的训练， 只要站在树下就一定能接住这棵树上掉落的苹果。同时，贝茜能够在两棵树之间 快速移动（移动时间远少于1分钟），因此当苹果掉落时，她必定站在两棵树其中的一棵下面。此外，奶牛不愿意不停地往返于两棵树之间，因此会错过一些苹果。苹果每分钟掉落一个，共T（1<=T<=1000）分钟，贝茜最多愿意移动W（1<=W<=30） 次。现给出每分钟掉落苹果的树的编号，要求判定贝茜能够接住的最多苹果数。 开始时贝茜在1号树下。

输入格式

第一行2个数，T和W。接下来的t行，每行一个数，代表在时刻t苹果是从1号苹果树还是从2号苹果树上掉下来的。

输出格式

对于每个测试点，输出一行，一个数，为奶牛最多接到的苹果的数量。

输入输出样例

输入 #1复制

7 2
2
1
1
2
2
1
1

输出 #1复制

6

---

思路

令dp[i][j][k]为奶牛在第i分钟移动j步，在第k颗树下能够接到的最多苹果数。k用0或1表示。

1.当奶牛不移动的时候

dp[i][j][k]=dp[i-1][j][k]+f

其中f来判断当前这棵树在第i分钟是否能掉落果子，也就是a[i]是否等于k+1（k+1表示的是树的编号），能则f=1，否则f=0。

2.当奶牛移动的时候

dp[i][j][k]=max(dp[i][j][k],dp[i-1][j-1][k^1]+f)

其中k^1表示取异或。

答案在代码中有解释。

#include <stdio.h>
#include <iostream>
using namespace std;
int dp[1001][31][2],a[1001],n,m,s,f;
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	register int i,j,k;
	cin>>n>>m;
	for(i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	for(i=1;i<=n;i++)
	{
		for(j=0;j<=m;j++)
		{
			for(k=0;k<=1;k++)
			{
				if(a[i]==k+1) f=1;
				else f=0;
				dp[i][j][k]=dp[i-1][j][k]+f;
				dp[i][j][k]=max(dp[i][j][k],dp[i-1][j-1][k^1]+f);
			}
		}
	}
	for(j=0;j<=min(n,m);j++)
	{
		if(dp[n][j][j%2]>s)//奶牛在最后1分钟，枚举移动次数j，移动奇数次肯定在2号苹果树，移动偶数次在1号苹果树 
		{
			s=dp[n][j][j%2];
		}
	}
	cout<<s<<endl;
	return 0;
}