AcWing 1058. 股票买卖 V

AcWing 1058. 股票买卖 V

本题多了一个冷却期的状态：我们设为\(f[i][2]\),
无货设为\(f[i][0]\), 有货设为\(f[i][1]\)。

定义一个冷却期感觉比y总的那个状态定义更好理解，尤其在初始化的时候。

初始化：f[0][0] = 0 其余为-0x3f3f3f3f。

\(f[i][0] = max(f[i - 1][0], f[i - 1][2])\)：表示无货可以从冷却期还有无货转移而来。
\(f[i][1] = max(f[i - 1][0] - w[i], f[i - 1][1])\)：表示有货可以从有货与无货转移而来。
\(f[i][2] = max(f[i][2], f[i - 1][1] + w[i])\)：表示冷却期只能从有货转移而来。

目标：\(max(无货，冷却期) = max(f[n][2], f[n][0])\)

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

using namespace std;

const int N = 1e5 + 10;
int f[N][3];
int n;
int w[N];

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> w[i];
    
    memset(f, -0x3f, sizeof f);
    f[0][2] = 0;
    for (int i = 1; i <= n; i++) {
        f[i][0] = max(f[i - 1][0], f[i - 1][2] - w[i]);
        f[i][1] = f[i - 1][0] + w[i];
        f[i][2] = max(f[i - 1][2], f[i - 1][1]);
    }
    
    cout << max(f[n][0], max(f[n][1], f[n][2])) << endl;
    
    return 0;
}