code39 股票的买卖时机问题3

119 阅读2分钟

题目描述

  • 给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
  • 设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。
  • 注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

举例

  • 示例1:
  • 输入:prices = [3,3,5,0,0,3,1,4]
  • 输出:6
  • 解释:在第 4 天(股票价格 = 0)的时候买入,在第 6 天(股票价格 = 3)的时候卖出,这笔交易所能获得利润 = 3-0 = 3 。
  • 随后,在第 7 天(股票价格 = 1)的时候买入,在第 8 天 (股票价格 = 4)的时候卖出,这笔交易所能获得利润 = 4-1 = 3 。
  • 示例 2:
  • 输入:prices = [1,2,3,4,5]
  • 输出:4
  • 解释:在第 1 天(股票价格 = 1)的时候买入,在第 5 天 (股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
  • 注意你不能在第 1 天和第 2 天接连购买股票,之后再将它们卖出。  
  • 因为这样属于同时参与了多笔交易,你必须在再次购买前出售掉之前的股票。
  • 示例 3:
  • 输入:prices = [7,6,4,3,1]
  • 输出:0
  • 解释:在这个情况下, 没有交易完成, 所以最大利润为 0。
  • 示例 4:
  • 输入:prices = [1]
  • 输出:0

思路:

  • 从左到右的尝试模型
  • 一定要在 i 号位置做完两次交易
  • onceBuy 做完一次交易并且减去第二次交易的买入价格的最好情况

code

public class Code39 {

    public static int maxProfit(int[] prices) {
        if (prices == null || prices.length <= 1) {
            return 0;
        }
        // 做完一次交易并且减去第二次交易的买入价格的最好情况
        int onceBuy = -prices[0];
        // 做完一次交易的最好情况
        int once = 0;
        // 最小值
        int min = prices[0];
        // 答案
        int ans = 0;
        for (int i = 1; i < prices.length; i++) {
            // 当前位置答案就是  做完一次交易并且减去第二次交易的买入价格的最好情况  + 当前的值
            ans = Math.max(onceBuy + prices[i], ans);
            // 更新最小值
            min = Math.min(min, prices[i]);

            // 在当前位置做完一次交易的最好情况
            once = Math.max(prices[i] - min, once);
            // 做完一次交易并且减去第二次交易的买入价格的最好情况 (当前位置买入,当前位置不买入)
            onceBuy = Math.max(once - prices[i], onceBuy);
        }
        return ans;
    }
}