刷题No22. best-time-to-buy-and-sell-stock-iii(买卖股票的最佳时机III)(java)【数组】

题目:

假设你有一个数组,其中第i个元素是某只股票在第i天的价格。
设计一个算法来求最大的利润。你最多可以进行两次交易。
注意:
你不能同时进行多个交易(即,你必须在再次购买之前出售之前买的股票)。
示例1:

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

示例2:

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

示例3:

输入: [7,6,4,3,1]

输出: 0

解释:该情况下没有进行交易,所以收益为0。

思路:贪心算法

  • 从左到右进行遍历。buy1表示组最便宜股票的价格,sell1表示初始时的利润
  • buy2表示第一支股票售卖之后,继续购买第二支股票之后,手里剩余的钱。sell2 表示交易两次之后的利润。

代码:

package com.company;

public class TestNo22 {
    public static void main(String[] args) {
        TestNo22 t = new TestNo22();
        int[] a = {3, 3, 5, 0, 0, 3, 1, 4};
        System.out.println(t.maxProfit(a));

    }
    public int maxProfit(int[] prices) {
        if (prices == null || prices.length <= 1) {
            return 0;
        }
        int buy1 = Integer.MIN_VALUE;
        int sell1 = 0;
        int buy2 = Integer.MIN_VALUE;
        int sell2 = 0;
        for(int i= 0 ;i<prices.length;i++){
            int curPrice = prices[i];
            //最便宜的股票价格
            buy1 = Math.max(buy1,-curPrice);
            //一次交易之后的最大收益
            sell1 = Math.max(sell1,curPrice+buy1);
            //之前已经进行了一次交易,这是在买入第二次股票之后的净利润
            buy2 = Math.max(buy2,sell1-curPrice);
            //二次交易之后的总收益
            sell2 = Math.max(sell2,buy2+curPrice);
        }
        return sell2;

    }
}
发布了46 篇原创文章 · 获赞 11 · 访问量 3582

猜你喜欢

转载自blog.csdn.net/qq_40664693/article/details/104236732