letcode123. 买卖股票的最佳时机 III

前面一题时卖股票的最佳时机Ⅰ
不知道为啥没有Ⅱ,Ⅰ是常规的简单dp,就直接写Ⅲ了。


尝试用股票Ⅰ的思路想,失败了,状态没有考虑全面。

由于可以买两股股票,状态变多了。
有着是否此时持有股票,此时已经卖出了几个股票。
可分为以下几种状态:
dp[i][0][0];此时状态为未持股票且一股都没有卖过。
dp[i][1][0];此时状态为持有一张股票且一股都没有卖过。而这一张股票可以是此时i的时候购进来的,也可以是在上一次股票就已经购买的。
dp[i][1][1];此时状态为持有一张股票且已经卖过了一次。由于最近的状态是持有的这张股票而不是已经卖过的那张,因此状态转换更上一项一样。
dp[i][0][1];此时状态为未持有股票且已经卖过了一次。此时就需要讨论这个被卖过的这股股票了,要么是此时刚卖的,要么就是之前卖掉的
dp[i][0][2];此时状态为未持有股票并且已经卖了两次了,最近卖的这次也是两种状态移来,一是此时刚卖的,二就是之前早已卖好。

ok所有状态都来了,接下里撸代码即可。
不想用三维的,其实都一样。
注意初始化

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len = prices.size();
        if(len < 2) return 0;
        int mi = -0x3f3f3f3f;
        vector<vector<int>>dp(len, vector<int>(5));
        dp[0][0] = 0;
        dp[0][3] = -prices[0];
        dp[0][2] = mi;
        dp[1][2] = mi;
        dp[0][1] = mi;
        dp[0][4] = mi;
        dp[1][4] = mi;
        for(int i=1; i<len; ++i){
            dp[i][0] = 0;
            dp[i][1] = max(dp[i-1][3]+prices[i], dp[i-1][1]);
            dp[i][2] = max(dp[i-1][4]+prices[i], dp[i-1][2]);
            dp[i][3] = max(dp[i-1][0]-prices[i], dp[i-1][3]);
            dp[i][4] = max(dp[i-1][1]-prices[i], dp[i-1][4]);
        }
        return max(0, max(dp[len-1][2], dp[len-1][1]));
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值