股票买卖IV（状态机模型）

最新推荐文章于 2025-06-04 00:33:14 发布

原创

最新推荐文章于 2025-06-04 00:33:14 发布 · 624 阅读

0 ·

CC 4.0 BY-SA版权

本文探讨了一种股票交易策略，基于动态规划和状态机模型，最多允许进行k次交易，并且每次交易只能持有一支股票。通过dp[i][j][0/1]来表示前i天交易j次且当前无/有股票的最大利润。策略包括考虑不持有股票时，可能是前一天无股票或卖出股票；持有股票时，可能是前一天已持有或买入股票。初始化所有状态为负无穷，第0次交易且不持有股票的状态值为0。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

思路：只能持有一个股票所以要么卖出要么买入，最多交易k次，那么我们就用dp[i][j][0/1]表示前i天交易j次当前没有/有股票的最大利润。如果第i天交易j次时手上没有股票那么就有2种可能

1.在前i-1天已经交易了j次没有股票了

2.在前i-1天是持有第j次交易的股票，然后在第i天吧股票的卖掉了

dp[i][j][0]=max(dp[i-1][j][0],dp[i-1][j][1]+a[i])

当第i天持有股票时也是2种状态

1.在前i-1天的时候已经持有了股票

2.在前i-1天的时候不持有股票然后在第i天买入了第j次交易的股票

dp[i][j][1]=max(dp[i-1][j-1][0]-a[i],dp[i-1][j][1])

初始化时先都置为-inf，在前N天的第0次交易且不持有股票时的值都是0.

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#include<iostream>
#include<vector>
#include<queue>
#include<stack>
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define SIS std::ios::sync_with_stdio(false)
#define space putchar(' ')
#define enter putchar('\n')
#define lson root<<1
#define rson root<<1|1
typedef pair<int,int> PII;
const int mod=1e9+7;
const int N=2e6+10;
const int M=1e3+