Apple Catching
|
题目链接:Problem - 2385
题目大意:John有2棵苹果树,每个时刻i在某棵树上会掉下来一个苹果,John想在限定的步数内获得最多的苹果,一开始John在苹果树1下面,问最多的苹果个数是多少。
题目分析:
典型的dp,每次更新求出最优解。
int dp[1005][35][2];//记录i时间在第j步在第k个树下能拿到的最大苹果数
每个点有二种可能(一种是上个点变步数走过来,一种是上个点不变步数走过来,然后满足条件可获得当前的苹果)
状态转移方程:dp[i][j][k] = max(dp[i-1][j-1][!k], dp[i-1][j][k]);
思路分析:
1)存图,初始化dp数组。
2)三重循环扫一遍dp数组,满足条件的更新,否则置为-1;
贴AC代码:
#include <stdio.h> //定义输入/输出函数
#include <limits.h> //定义各种数据类型最值常量
#include <math.h> //定义数学函数
#include <stdlib.h> //定义杂项函数及内存分配函数
#include <string.h> //字符串处理
#include <algorithm>//算法
#include <queue>//队列
#include <stack>//栈
#include <vector>
using namespace std;
int n, m, ans;
int map[1005];//记录苹果掉落位置
int dp[1005][35][2];//记录i时间在第j步在第k个树下能拿到的最大苹果数
int main()
{
while (scanf("%d", &n) != EOF && n){
scanf("%d", &m);
for (int i = 1; i <= n; i++){
scanf("%d", &map[i]);
if (map[i] == 2)
map[i] = 0;
//printf("%d ",map[i]);
}
map[0] = 1;
memset(dp, -1, sizeof(dp));
if (map[1] == 1) dp[1][0][1] = 1;
if (map[1] == 0) {dp[1][0][1] = 0;dp[1][1][0] = 1;}
for (int i = 2; i <= n; i++){ //扫时间点
for (int j = 0; j <= m; j ++){ //扫步数
for (int k = 0; k < 2; k++){ //扫位置
if (dp[i-1][j-1][!k] == -1 && dp[i-1][j][k] == -1) continue;
dp[i][j][k] = max(dp[i-1][j-1][!k], dp[i-1][j][k]);
if (k == map[i])
dp[i][j][k] ++;
//printf("(%d,%d,%d) = %d\n", i, j, k, dp[i][j][k]);
}
}
}
ans = 0;
for (int i = 1; i <= n; i++)
for (int k = 0; k <2; k++)
if (ans < dp[i][m][k])
ans = dp[i][m][k];
printf("%d\n", ans);
}
}