POJ 2385 Apple Catching

POJ 2385题目大意如下：

（这个题，去年学院选拔考试考过，然而并不会做，今天又遇见，煎熬两小时。。。）

FJ的牛喜欢吃苹果，有两棵苹果树，每隔一分钟，有且仅有一棵苹果树会掉下一个苹果，不计牛的运动时间，算出牛最多能吃多少个苹果（限定了牛在两棵树之间的最大移动次数）。

典型的DP问题，之前想过DP树，但是之后不知道怎么化简，转而考虑状态方程。对于第i分钟牛的第j次移动有以下规律：

1）牛在第i－1分钟的时候已经移动了j次；

2）牛在第i－1分钟的时候已经移动了j－1次，第i分钟移动了j次；

所以状态方程如下：

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

dp表示的是第i分钟调转j次吃到的最大苹果数，但是这个式子用来更新结果，显然是不对的，因为在第j次转移的时候，还没有考虑到此时牛所在的那棵树到底掉没掉苹果，也就是说，还没有给dp加1，但是这种相加要分情况：

1）如果j为偶数，显然，牛应该在第1棵树下，如果掉了苹果那么以上dp[i][j] += 1；

2）如果j为奇数，显然，牛应该在第2棵树下，如果掉了苹果那么以上dp[i][j] += 1；

这就OK了，以下是代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>

using namespace std;
const int maxn = 1000;

int T, W;
int tree[maxn + 1] = {1};
int dp[maxn + 2][35];

void solve() {
    int trans = 0;
    memset(dp, 0, sizeof(dp));
    for (int i = 1; i <= T; i++) {
        if (tree[i] != tree[i - 1]) trans++;
        int M = trans < W ? trans : W;
        for (int j = 0; j <= M; j++) {
            if (j == 0) dp[i][0] = dp[i - 1][0] + tree[i] % 2;
            else {
                dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - 1]);
                if ((tree[i] == 2 && j % 2) || (tree[i] == 1 && j % 2 == 0)) dp[i][j] += 1;
            }
        }
    }
    printf("%d\n", dp[T][W]);
}

int main(int argc, const char * argv[]) {
    // insert code here...
    scanf("%d %d", &T, &W);
    for (int i = 1; i <= T; i++) scanf("%d", &tree[i]);
    solve();
    return 0;
}

Accept 836K／16MS