LeetCode 力扣 123. 买卖股票的最佳时机 III

题目描述(困难难度)

依旧是买卖股票的延伸,但比 121 题122 题 难度高了不少。这道题的意思是,给一个数组代表股票每天的价格。你最多可以买入卖出两次,但只有卖出了才可以再次买入,求出最大的收益是多少。

解法一

参考 这里

开始的想法是求出收益第一高和第二高的两次买卖,然后加起来。对于普通的情况是可以解决的,但是对于下边的情况

1 5 2 8 3 10

第一天买第二天卖,第三天买第四天卖,第五天买第六天卖,三次收益分别是 467,最高的两次就是 6 + 7 = 13 了,但是我们第二天其实可以不卖出,第四天再卖出,那么收益是 8 - 1 = 7,再加上第五天买入第六天卖出的收益就是 7 + 7 = 14了。

所以当达到了一个高点时不一定要卖出,所以需要考虑的情况就很多了,不能像 121 题122 题 那样简单的考虑了。那只能朝着动态规划思路想了。

动态规划关键就是数组定义和状态转移方程了。

按最简单的动态规划的思路想,用 dp[i]表示前i天的最高收益,那么 dp[i+1] 怎么根据 dp[i] 求出来呢?发现并不能求出来。

我们注意到我们题目是求那么多天最多交易两次的最高收益,还有一个最多交易次数的变量,我们把它加到数组中再试一试。

dp[i][k] 表示前i天最多交易k次的最高收益,那么 dp[i][k] 怎么通过之前的解求出来呢?

首先第 i 天可以什么都不操作,今天的最高收益就等于昨天的最高收益

dp[i][k] = dp[i-1][k]

此外,为了获得更大收益我们第 i 天也可以选择卖出,既然选择卖出,那么在0i-1 天就要选择一天买入。多选择了一次买入,那在买入之前已经进行了 k-1 次买卖。

在第 0 天买入,收益就是 prices[i] - prices[0]

在第 1 天买入,收益就是 prices[i] - prices[1] + dp[0][k-1],多加了前一天的最大收益

在第 2 天买入,收益就是 prices[i] - prices[2] + dp[1][k-1],多加了前一天的最大收益

...

在第 j 天买入,收益就是 prices[i] - prices[j] + dp[j-1][k-1],多加了前一天的最大收益

上边的每一种可能选择一个最大的,然后与第i天什么都不操作比较,就是dp[i][k]的值了。

当然上边的推导已经可以写代码了,但为了最后的代码更加简洁(写完代码后发现的),我们可以再换一下状态转移方程。真的只是为了简洁,时间复杂度和空间复杂度上不会有影响。

为了获得更大收益我们第 i 天也可以选择卖出,既然选择卖出,那么在0i-1 天就要选择一天买入。

我们也可以选择0i天中选择一天买入,因为第 i 天买入,第 i天卖出对最后的收益是没有影响的。

在第 j 天买入,收益就是 prices[i] - prices[j] + dp[j-1][k-1],多加了前一天的最大收益

我们多加了前一天的最大收益,我们也可以改成加当前天的最大收益。

在第 j 天买入,收益就是 prices[i] - prices[j] + dp[j][k-1]

不严谨的想一下,如果第 j 天就是最后我们要选择的买入点,它使得最后的收益最高,dp[j][k-1]dp[j-1][k-1] 一定是相等的。因为第 j 天一定是一个低点而第 j - 1 天是个高点,第 j 天为了得到更高收益肯定选择不操作,所以和第 j - 1 天的收益是一样的,所以改了状态转移方程,最后求出的最高解还是一致的。

综上,最后的状态转移方程就是

dp[i][k] = Max(dp[i-1][k],(prices[i] - prices[0] + dp[0][k-1]),(prices[i] - prices[1] + dp[1][k-1])...(prices[i] - prices[i] + dp[i][k-1]))

也就是

dp[i][k] = Max(dp[i-1][k],prices[i] - prices[j] + dp[j][k-1])j0 - i

prices[i] - prices[j] + dp[j][k-1] 也可以看做, prices[i] - (prices[j] - dp[j][k-1]) ,为了求这个表达式的最大值,我们可以找prices[j] - dp[j][k-1]的最小值。

而初始条件对于k 等于 0 的情况,收益就是 0 了。

还有前 0 天的最大收益也是 0 ,也就是dp[0][k]是 0 。由于下标是从0开始的,这里的前0天其实就是第一天。

因为初始条件的结果都是0,数组初始化后就是 0 ,所以不需要特殊处理。

public int maxProfit(int[] prices) {
    if (prices.length == 0) {
        return 0;
    }
    int K = 2;
    int[][] dp = new int[prices.length][K + 1];
    for (int k = 1; k <= K; k++) {
        for (int i = 1; i < prices.length; i++) { 
            int min = Integer.MAX_VALUE;
            //找出第 0 天到第 i 天 prices[buy] - dp[buy][k - 1] 的最小值
            for (int buy = 0; buy <= i; buy++) { 
                min = Math.min(prices[buy] - dp[buy][k - 1], min);
            }
            //比较不操作和选择一天买入的哪个值更大
            dp[i][k] = Math.max(dp[i - 1][k], prices[i] - min);
        }
    }
    return dp[prices.length - 1][K];
}

找第 jprices[buy] - dp[buy][k - 1]的最小值的时候,我们考虑了 prices[0] - dp[0][k - 1]prices[1] - dp[1][k - 1]prices[2] - dp[2][k - 1]...,找第 j + 1prices[buy] - dp[buy][k - 1]的最小值的时候,我们又会从头考虑 prices[0] - dp[0][k - 1]prices[1] - dp[1][k - 1]prices[2] - dp[2][k - 1]...,所以其实没必要每次从头考虑,我们只需要把之前的结果保存起来,然后再和新加入的 prices[j+1] - dp[j+1][k - 1] 比较就可以了。

public int maxProfit(int[] prices) {
    if (prices.length == 0) {
        return 0;
    }
    int K = 2;
    int[][] dp = new int[prices.length][K + 1];
    for (int k = 1; k <= K; k++) {
        int min = prices[0];
        for (int i = 1; i < prices.length; i++) {
            //找出第 1 天到第 i 天 prices[buy] - dp[buy][k - 1] 的最小值 
            min = Math.min(prices[i] - dp[i][k - 1], min); 
            //比较不操作和选择一天买入的哪个值更大
            dp[i][k] = Math.max(dp[i - 1][k], prices[i] - min);
        }
    }
    return dp[prices.length - 1][K];
}

此时按照动态规划的套路,结合代码和下边的图。

根据代码,我们是固定 k 然后一列一列更新 dp。而更新当前列只需要前一列的信息,所以不需要二维数组,只需要一个一维数组。但是注意到最外层的 for 循环是一个常数次,所以我们可以把两层循环内外颠倒下,可以更好的进行空间复杂度的优化。

public int maxProfit(int[] prices) {
    if (prices.length == 0) {
        return 0;
    }
    int K = 2;
    int[][] dp = new int[prices.length][K + 1];
    int min[] = new int[K + 1];
    for (int i = 1; i <= K; i++) {
        min[i] = prices[0];
    }
    for (int i = 1; i < prices.length; i++) {
        for (int k = 1; k <= K; k++) {
            min[k] = Math.min(prices[i] - dp[i][k - 1], min[k]);
            dp[i][k] = Math.max(dp[i - 1][k], prices[i] - min[k]);
        }
    }
    return dp[prices.length - 1][K];
}

再结合图看,此时我们就是一行一行的更新了,对于每一列都有一个 min 所以我们多了 min 数组。现在让我们将二维数组 dp 改成一维数组。

public int maxProfit(int[] prices) {
    if (prices.length == 0) {
        return 0;
    }
    int K = 2;
    int[] dp = new int[K + 1];
    int min[] = new int[K + 1];
    for (int i = 1; i <= K; i++) {
        min[i] = prices[0];
    }
    for (int i = 1; i < prices.length; i++) {
        for (int k = 1; k <= K; k++) {
            min[k] = Math.min(prices[i] - dp[k - 1], min[k]);
            dp[k] = Math.max(dp[k], prices[i] - min[k]);
        }
    }
    return dp[K];
}

由于 K 是一个常数,所以我们的 min 数组和 dp 数组都可以分别当成两个变量。

public int maxProfit(int[] prices) {
    if (prices.length == 0) {
        return 0;
    } 
    int dp1 = 0;
    int dp2 = 0;
    int min1 = prices[0];
    int min2 = prices[0];
    for (int i = 1; i < prices.length; i++) {
            min1 = Math.min(prices[i] - 0, min1);
            dp1 = Math.max(dp1, prices[i] - min1);

            min2 = Math.min(prices[i] - dp1, min2);
            dp2 = Math.max(dp2, prices[i] - min2);
    }
    return dp2;
}

如果结合一步一步的优化,最后这个代码也就很好的能解释通了。

解法二

再分享个利用状态机的 解法,虽然不容易想到,但真的太强了,上次用状态机还是 65 题

每天我们其实是有四个状态,买入当前价格的股票,以当前价格的股票卖出。第二次买入股票,第二次卖出股票。

s0代表初始状态,初始时钱是 0s1代表第一次买入后当前的钱,s2代表第一次卖出后当前的前,s3代表第二次买入后当前的钱,s4代表第二次卖出后当前的钱。

然后我们只需要更新每天的这四个状态即可。

int maxProfit(vector<int>& prices) {
    if(prices.empty()) return 0;
    //进行初始化,第一天 s1 将股票买入,其他状态全部初始化为最小值
    int s1=-prices[0],s2=INT_MIN,s3=INT_MIN,s4=INT_MIN;

    for(int i=1;i<prices.size();++i) {            
        s1 = max(s1, -prices[i]); //买入价格更低的股
        s2 = max(s2, s1+prices[i]); //卖出当前股,或者不操作
        s3 = max(s3, s2-prices[i]); //第二次买入,或者不操作
        s4 = max(s4, s3+prices[i]); //第二次卖出,或者不操作
    }
    return max(0,s4);
}

解法一比较常规,但是这个动态规划难在了我们考虑了两个变量,相比于之前的动态规划不容易想到。

更多详细通俗题解详见 leetcode.wang

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,287评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,346评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,277评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,132评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,147评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,106评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,019评论 3 417
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,862评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,301评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,521评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,682评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,405评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,996评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,651评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,803评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,674评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,563评论 2 352

推荐阅读更多精彩内容