acwing 1057 股票买卖IV

最新推荐文章于 2022-07-25 01:54:06 发布

(xsj)

最新推荐文章于 2022-07-25 01:54:06 发布

阅读量142

点赞数

分类专栏： DP 文章标签：动态规划股票交易状态转移最大利润算法

本文链接：https://blog.csdn.net/qq_44791484/article/details/115408127

版权

DP 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

题面

在这里插入图片描述

题解(状态机)

f[i] [j] [0] : 从前 i 天中交易完成 j 次并且现在是没有买入股票的
f[i] [j] [1] : 从前 i 天中交易完成 j 次并且现在是买入股票的

在这里插入图片描述

代码

#include<bits/stdc++.h>

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

int n, m;
int w[N];
int f[N][M][2];

int main() {

    cin >> n >> m;
    for (int i = 1; i <= n; i++) cin >> w[i];
    //不合法状态初始化成-INF(求最大值更新为—INF,求最小值更新成INF)
    // f[i][0][1] 前i天没有进行交易但手中却有股票是不合法的
    memset(f, -0x3f, sizeof f);
    //前i天交易,交易完成次数为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]);
        }
    }
    int res = 0;
    //我们发现买入不卖一定不是最优解,所以不用枚举f[i][j][1]的状态
    for (int i = 0; i <= m; i++) res = max(res, f[n][i][0]);
    cout << res << endl;
    return 0;
}