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]));
    }
};
发布了75 篇原创文章 · 获赞 26 · 访问量 7651

猜你喜欢

转载自blog.csdn.net/qq_40962234/article/details/104732221