【每日算法Day 67】买卖股票的最佳时机 III

LeetCode 123. 买卖股票的最佳时机 III (opens new window)

题目描述

给定一个数组，它的第 i 个元素是一支给定的股票在第 i 天的价格。

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

注意: 你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

示例

输入: [3,3,5,0,0,3,1,4]
输出: 6
解释: 在第 4 天（股票价格 = 0）的时候买入，在第 6 天（股票价格 = 3）的时候卖出，这笔交易所能获得利润 = 3-0 = 3 。
     随后，在第 7 天（股票价格 = 1）的时候买入，在第 8 天 （股票价格 = 4）的时候卖出，这笔交易所能获得利润 = 4-1 = 3 。

解题思路

这一题当k比较大时可能会想到使用使用动态规划，这里由于k=2，可以结合第一题中的贪心思路，用第一笔交易赚到的钱抵消第二次购买的钱。即在寻找可能的第二次交易前的最低价时，用当天的股价减去已知的在这之前完成第一次交易后的最大利润得到第二次买入时的最低价，再从第二次交易的角度寻找最大利润。

public int maxProfit(int[] prices) {
    if (prices == null || prices.length <=1) return 0;
    int min1 = Integer.MAX_VALUE, min2 = Integer.MAX_VALUE;
    int maxProfit1 = 0, maxProfit2 = 0;
    for (int price : prices) {
        min1 = Math.min(min1, price);
        maxProfit1 = Math.max(maxProfit1, price - min1);
        min2 = Math.min(min2, price - maxProfit1);
        maxProfit2 = Math.max(maxProfit2, price - min2);
    }
    return maxProfit2;
}