Dp-状态机模型：股票买卖Ⅴ 三种状态

题目链接：https://www.acwing.com/problem/content/1060/


初始化：只有入口f[0][2]合法，初始化成0；f[0][1]和f[0][0]都是非法状态，初始化成负无穷。

结果：出口，max(f[n][1], f[n][2])

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

using namespace std;

const int N = 100010, INF = 0x3f3f3f3f;

int n;
int w[N];
int f[N][3];

int main()
{
    scanf("%d", &n);

    for (int i = 1; i <= n; i ++ ) scanf("%d", &w[i]);

    // 合法状态为0，不合法状态负无穷
    // 初始化：只有入口f[0][2]合法，初始化成0；
    // f[0][1]和f[0][0]都是非法状态，初始化成负无穷。

    f[0][0] = f[0][1] = -INF, 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]);
    }

    printf("%d\n", max(f[n][1], f[n][2]));

    return 0;
}