[状态机dp] 股票买卖 IV(状态机模型)

0. 前言

以前写的垃圾博文… [Hdp] lc188. 买卖股票的最佳时机 IV(状态机模型dp+线性dp+空间优化)。

1. 状态机模型

1057. 股票买卖 IV

状态机模型

注意：

• 一次交易指的是买入后再卖出，算作一次交易。光是股票买入，交易数量 j 是不做改变的。得从状态定义出发理解状态转移方程
• 初始化只可能为 0 或者 -INF，分别对应当前状态为有效状态可以转移。当前状态为无效状态不可转移过来
• 输出答案时，需要遍历所有操作数再输出。且手里不能有股票

代码：

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

using namespace std;

const int N = 1e5+5, M = 105, INF = 0x3f3f3f3f;

int n, k;
int a[N];
int f[N][M][2];

int main() {
    cin >> n >> k;
    
    for (int i = 1; i <= n; ++i) cin >> a[i];
    
    memset(f, 0xcf, sizeof f);
    for (int i = 0; i <= n; ++i) f[i][0][0] = 0;
    
    for (int i = 1; i <= n; ++i) 
        for (int j = 1; j <= k; ++j) {
            f[i][j][0] = max(f[i - 1][j][1] + a[i], f[i - 1][j][0]);
            f[i][j][1] = max(f[i - 1][j - 1][0] - a[i], f[i - 1][j][1]);
        }
    
    int res = 0;
    for (int i = 0; i <= k; ++i) res = max(res, f[n][i][0]);    // f[n][i][0]是完整交易
    cout << res << endl;
    
    return 0;
}

---

另一种状态初始化的方式：

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        int n = prices.size();
        int f[n][k + 1][2]; memset(f, 0xcf, sizeof(f)); // 初始化负无穷大，非法值
        for (int i = 0; i < n; i ++ ) f[i][0][1] = 0;  // 手里没有股票，且没操作过时，不赚不亏
        f[0][1][0] = -prices[0];
        // for (int i = 0; i < n; i ++ ) {
        //     if (i == 0) f[i][1][0] = -prices[i]; // 手里有股票，仅买入，算作一次操作
        //     else f[i][1][0] = max(f[i - 1][1][0], -prices[i]);
        // }
        for (int i = 1; i < n; i ++ ) {
            for (int j = 1; j <= k; j ++ ) {
                f[i][j][0] = max(f[i - 1][j][0], f[i - 1][j - 1][1] - prices[i]); // 手里有股票，持有 或 买入，开启一次操作
                f[i][j][1] = max(f[i - 1][j][1], f[i - 1][j][0] + prices[i]); // 手里没股票，持有 或 卖出，不是一次新操作
            }
        }

        int res = 0;
        for (int i = 0; i <= k; i ++ ) res = max(res, f[n - 1][i][1]);
        return res;
    }
};

思考：这个 n+1 不是必要的，而 k+1 却是必要的。因为 k 取值在 0~k，代表一次都不操作，操作1，操作… 操作 k 次。

再降 n+1 到 n 的时候，顺便把 k+1 也降了，然后一顿报错…淦。