买卖股票的最佳时机(交易次数限制)

买卖股票的最佳时机(交易次数限制)

给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。

状态转移方程

buy[i][j]=max{buy[i−1][j],sell[i−1][j]−price[I]}
sell[i][j]=max{sell[i−1][j],buy[i−1][j−1]+price[i]}
需要注意卖出股票操作时,需要从购买次数 - 1的状态转移。
确定初始状态也很重要,buy[0][0] 代表第一天便买入的状态,而buy[i][0]表示1~n-1天每天卖出次数为0的状态,次状态只需要为买入前i天的最小价格的股票即可。
buy[0][j]为第一天买卖次数为1 ~ k的状态,不能从此状态转移,所以全部赋为0。
最后结果为最后一天,对所有卖出次数求最大值。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size(),k = 2;
		
		//表示n天进行了k次交易
        vector<vector<int>> buy(n, vector<int>(k + 1));
        vector<vector<int>> sell(n, vector<int>(k + 1));
		
        buy[0][0] = -prices[0];
        sell[0][0] = 0;
        for (int i = 1; i <= k; ++i) {
            buy[0][i] = sell[0][i] = INT_MIN / 2;
        }

        for (int i = 1; i < n; ++i) {
            buy[i][0] = max(buy[i - 1][0], sell[i - 1][0] - prices[i]);
            for (int j = 1; j <= k; ++j) {
                buy[i][j] = max(buy[i - 1][j],sell[i - 1][j] - prices[i]);
                sell[i][j] = max(buy[i - 1][j - 1] + prices[i],sell[i - 1][j]);  
            }
        }

        return *max_element(sell[n - 1].begin(), sell[n - 1].end());
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值