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

https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-iii/
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。

思路:动态规划
由于只能完成两笔交易,需要一维来记录交易次数。
第一笔交易:
第i天不持有股票,那么可能第i - 1天也不持有股票,也可能第i - 1天持有股票,而在i天时卖出
第i天持有股票,那么可能第i - 1天持有股票,也可能第i - 1天不持有股票,在第i天时买入
状态转移方程为:
dp[i][0][0] = Math.max(dp[i - 1][0][0], dp[i - 1][0][1] + prices[i]);
dp[i][0][1] = Math.max(dp[i - 1][0][1], 0 - prices[i]);
第二笔交易:
第i天不持有股票,那么可能第i - 1天也不持有股票,也可能第i - 1天持有股票,而在i天时卖出
第i天持有股票,那么可能第i - 1天持有股票,也可能第i - 1天不持有股票,已有第一笔交易的利润,在第i天时买入
状态转移方程:
dp[i][1][0] = Math.max(dp[i - 1][1][0], dp[i - 1][1][1] + prices[i]);
dp[i][1][1] = Math.max(dp[i - 1][1][1], dp[i - 1][0][0] - prices[i]);

var maxProfit = function(prices) {
    const len = prices.length;
    let dp = new Array(len).fill(0).map( () => new Array(2).fill(0).map( () => new Array(2).fill(0) ) );

    dp[0][0][0] = 0;
    dp[0][0][1] = -prices[0];
    dp[0][1][0] = 0;
    dp[0][1][1] = -prices[0];

    for (let i = 1; i < len; i++) {
        dp[i][0][0] = Math.max(dp[i - 1][0][0], dp[i - 1][0][1] + prices[i]);
        dp[i][0][1] = Math.max(dp[i - 1][0][1], 0 - prices[i]);

        dp[i][1][0] = Math.max(dp[i - 1][1][0], dp[i - 1][1][1] + prices[i]);
        dp[i][1][1] = Math.max(dp[i - 1][1][1], dp[i - 1][0][0] - prices[i]);
    }

    return dp[len - 1][1][0];
};

状态压缩:

var maxProfit = function(prices) {
    const len = prices.length;

    let sell1 = 0;
    let buy1 = -prices[0];
    let sell2 = 0;
    let buy2 = -prices[0];

    for (let i = 1; i < len; i++) {
        let temp = sell1;
        sell1 = Math.max(sell1, buy1 + prices[i]);
        buy1 = Math.max(buy1, 0 - prices[i]);

        sell2 = Math.max(sell2, buy2 + prices[i]);
        buy2 = Math.max(buy2, temp - prices[i]);
    }

    return sell2;
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值