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;
};

你可能感兴趣的:(leetCode刷题,leetcode,算法,职场和发展)