动态规划状态机模型：股票买卖I

题目描述

给定一个长度为 $N$ 的数组，数组中的第 $i$ 个数字表示一个给定股票在第 $i$ 天的价格。

设计一个算法来计算你所能获取的最大利润，你最多可以完成 $k$ 笔交易。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。一次买入卖出合为一笔交易。

输入格式

第一行包含整数 $N$ 和 $k$，表示数组的长度以及你可以完成的最大交易数量。

第二行包含 $N$ 个不超过 $10000$ 的正整数，表示完整的数组。

输出格式

输出一个整数，表示最大利润。

数据范围

$1\le N\le 10^5$,
$1\le k\le 100$

输入样例1

3 2
2 4 1

输出样例1

2

输入样例2

6 2
3 2 6 5 0 3

输出样例2

7

样例解释

样例1： 在第 1 天 (股票价格 = 2) 的时候买入，在第 2 天 (股票价格 = 4) 的时候卖出，这笔交易所能获得利润 = 4-2 = 2 。

样例2： 在第 2 天 (股票价格 = 2) 的时候买入，在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。随后，在第 5 天 (股票价格 = 0) 的时候买入，在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3。共计利润 4+3 = 7。

算法思想

根据题目描述，在$N$天里完成最多 $k$ 笔交易的前提下，计算所能获取的最大利润。可以将每一天作为一个阶段进行处理，但对每一个阶段来说，都可以进行股票买卖，因此可以借助状态机的思想，将每天的状态细分为手中没有股票和持有股票两种状态，如下图所示：

上述状态机描述了4种符合条件的状态转换：

• 没有股票（空仓）-> 没有股票（空仓），继续空仓
• 持有股票（持仓）-> 没有股票（空仓），清仓卖出
• 持有股票（持仓）-> 持有股票（持仓），继续持有
• 没有股票（空仓）-> 持有股票（持仓），开仓买入

因此，可以使用该状态机模型定义每个阶段的状态，并进行状态计算。

状态表示

• f[i][j][0]表示前i天，且正在进行第j笔交易时，手中没有股票时的最大收益
• f[i][j][1]表示前i天，且正在进行第j笔交易时，手中持有股票时的最大收益

状态计算

• 没有股票（空仓）：f[i][j][0] = max(f[i - 1][j][0], f[i - 1][j][1] + w[i]) ，表示取继续空仓，或者是卖出上一阶段持有股票时获得的收益最大值
• 持有股票（持仓）：f[i][j][1] = max(f[i - 1][j][1], f[i - 1][j - 1][0] - w[i])，表示取继续持仓，或者是买入股票时获得的收益最大值

初始状态

f[i][0][0] = 0表示不进行任何交易时，不持有股票（空仓）收益为0。其它状态均为-INF，即不合理状态。

时间复杂度

状态数为：$n\times m\times 2$
转移计算：$1$
时间复杂度为：$O(1)$

代码实现

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N = 100010, M = 110;
//f[i][j][0]表示前i天，且正在进行第j笔交易时，手中没有股票时的最大收益
//f[i][j][1]表示前i天，且正在进行第j笔交易时，手中有股票时的最大收益
int w[N], f[N][M][2];
int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    
    for(int i = 1; i <= n; i ++) scanf("%d", &w[i]);
    
    //初始状态，负无穷，表示非法状态
    memset(f, -0x3f, sizeof f);
    //不进行任何交易时，不持有股票（空仓）收益为0
    //注意从0开始
    for(int i = 0; i <= n; i ++) f[i][0][0] = 0;
    
    for(int i = 1; i <= n; i ++)
        for(int j = 1; j <= m; j ++)
        {
            //不持有股票（空仓）
            f[i][j][0] = max(f[i - 1][j][0], f[i - 1][j][1] + w[i]);
            //持有股票（持仓）
            f[i][j][1] = max(f[i - 1][j][1], f[i - 1][j - 1][0] - w[i]);
        }
    //打擂台求前n天，在进行不超过m次交易时的最大收益
    int res = 0;
    for(int i = 0; i <= m; i ++) res = max(res, f[n][i][0]);
    printf("%d\n", res);
    
    return 0;
}