acwing 1055. 股票买卖 II

题面

题解(状态机模型)

f[i][0] : 第 i 天手中没有股票
f[i][1] : 第 i 天手中有股票
f[0][0] =0 （第0天手中没有股票）f[0][1]= -INF(非法状态，题中让求最大值，所以初始化为最小值)

代码

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

using namespace std;
const int N = 1e5 + 10;
const int INF = 0x3f3f3f3f;

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

int main() {

    cin >> n;
    for (int i = 1; i <= n; i++) cin >> w[i];
    f[0][0] = 0;
    f[0][1] = -INF;
    for (int i = 1; i <= n; i++) {
        f[i][0] = max(f[i - 1][0], f[i - 1][1] + w[i]);
        f[i][1] = max(f[i - 1][1], f[i - 1][0] - w[i]);
    }
    cout << max(f[n][0], f[n][1]) << endl;

    return 0;
}