POJ 2385 - Apple Catching dp水题

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);
    }
}