题目描述
很少有人知道奶牛爱吃苹果。农夫约翰的农场上有两棵苹果树(编号为 1 和 2),每一棵树上都长满了苹果。奶牛贝茜无法摘下树上的苹果,所以她只能等待苹果从树上落下。但是,由于苹果掉到地上会摔烂,贝茜必须在半空中接住苹果(没有人爱吃摔烂的苹果)。贝茜吃东西很快,所以她接到苹果后仅用几秒钟就能吃完。
每一分钟,两棵苹果树其中的一棵会掉落一个苹果。贝茜已经过了足够的训练,只要站在树下就一定能接住这棵树上掉落的苹果。同时,贝茜能够在两棵树之间快速移动(移动时间远少于 1 分钟),因此当苹果掉落时,她必定站在两棵树其中的一棵下面。此外,奶牛不愿意不停地往返于两棵树之间,因此会错过一些苹果。
苹果每分钟掉落一个,共T(1<=T<=1000)分钟贝茜最多愿意移动W(1<=W<=30)次。现给出每分钟掉落苹果的树的编号,要求判定贝茜能够接住的最多苹果数。开始时贝茜 在 1 号树下。
输入
第 1 行:由空格隔开的两个整数:T 和 W
第 2..T+1 行:1 或 2(每分钟掉落苹果的树的编号)
输出
第一行:在贝茜移动次数不超过 W 的前提下她能接到的最多苹果数
样例输入
7 2
2
1
1
2
2
1
1
样例输出
6
这是一道递推题,思想是动态规划设f(i,j)是第i个时间转换j次的最大值,那么我们可以搜索最后状态的f(i,1~n),下面是主要代码:
for(int i = 1; i <= n; ++i) {
f[i][0] = (a[i] == 1) + f[i - 1][0];
for (int j = 1; j <= m; ++j)
f[i][j] = (a[i] == j % 2 + 1) + max(f[i - 1][j - 1], f[i - 1][j]);
}
int Max = INT_MIN;
for (int i = 0; i <= n; ++i)
if (f[n][i] > Max)
Max = f[n][i]; 这就是这一道动态规划
|