CF 358D Dima and Hares

题目链接：洛谷 CF 358D Dima and Hares CF 358D Dima and Hares

算法标签: DP、思维

题目描述：

\(N\)个物品排成一排,按照一定顺序将所有物品都拿走，如果拿走某个物品时相邻两个物品都没有被拿过，那么得到的价值为\(a_i\)；如果相邻的两个物品有一个被拿过（左右无所谓），那么得到的价值为\(b_i\)；如果相邻的两个物品都被拿走了，那么对应价值为\(c_i\)。问能够获得的最高价值为多少。

题解：

看到题之后，第一反应就是贪心或者DP。仔细看一看，果断pass掉贪心的想法。

之后我们会发现这道题的DP有一些很难逾越的坑（导致我第一份代码写了140行还假了…………）：

每一个点的决策会改变前后两点的答案，前后两点的决策也会影响到当前点的答案
第一个点和最后一个点只能选择\(a\)或者\(b\)。

对于第一个问题，我们就要考虑如何搞方程才能满足这个条件。

如果每次都考虑当前点为中间点的话，的确很直接，很清晰，但是怎么同时对前后更改，（如果前后更改还会涉及到其它点的更改），这显然是实现不了的。

那么我们更换思路，考虑每一次遍历到的当前点是三个点中的最后一个，上图理解：

CF 358D Dima and Hares p1.png

这样我们可以搞出来一个DP状态：

\(dp[i][1]\)表示先选择\(i - 1\)，后选择\(i\)，可以得到的前\(i - 1\)的最大答案。

\(dp[i][0]\)表示先选择\(i\)，后选择\(i-1\)可以得到前\(i - 1\)的最大答案。

那么转移方程：

\(dp[i][1] = max(dp[i - 1][1] + b[i-1], dp[i - 1][0] + a[i-1])\)

\(dp[i][0] = max(dp[i - 1][1] + c[i-1], dp[i - 1][0] + b[i-1])\)

上图理解：

CF 358D Dima and Hares p2.png

这样第一个问题就解决了，那么怎么解决第二个问题：\(1\)号点和\(n\)号点。

首先因为每次取最大值，不妨设所有的\(dp[~][~]~=~-inf\)。

之后对于第一个点，由于它前边没有点，所以一定会先取这个点：\(dp[1][0]=0\)

对于最后一个点，由于他后边没有点，那么我们假设后边有一个点（由于状态当中\(dp[i]\)存放的是\(1 \sim i-1\)的答案），并且对于这个多出来的点来说，一定是先取\((n+1)-1\)这个点，所以在第\(n\)个点的答案自然就是\(dp[n+1][1]\)

现在状态有了，初值有了，转移方程有了，答案也有了，直接写dp就解决了。

AC代码

#include <bits/stdc++.h>

using namespace std;

#define setI(x) freopen(x".in", "r", stdin);

#define setO(x) freopen(x".out", "w", stdout);

#define setIO(x) setI(x) setO(x);

const int N = 100100;

typedef long long ll;

int n, a[N], b[N], c[N];

ll dp[N][2];

void rd() {
    for (int i = 1; i <= n; i ++ ) {
        scanf("%d", &a[i]);
    }
    for (int i = 1; i <= n; i ++ ) {
        scanf("%d", &b[i]);
    }
    for (int i = 1; i <= n; i ++ ) {
        scanf("%d", &c[i]);
    }
}

int main() {
    // setIO("cow");
    scanf("%d", &n);
    rd();
    memset(dp, -0x3f, sizeof dp);
    dp[1][0] = 0;
    for (int i = 2; i <= n + 1; i ++ ) {
        dp[i][0] = max(dp[i - 1][0] + b[i - 1], dp[i - 1][1] + c[i - 1]);
        dp[i][1] = max(dp[i - 1][0] + a[i - 1], dp[i - 1][1] + b[i - 1]);
    }
    cout << dp[n + 1][1] << endl;
    return 0;
}