赞
踩
Description: Say you have an array for which the ith element is the price of a given stock on day i. Design an algorithm to find the maximum profit. You may complete at most two transactions. Note: You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).
题意:用一个数组表示股票每天的价格,数组的第i个数表示股票在第i天的价格。最多交易两次,手上最多只能持有一支股票,求最大收益。
分析:动态规划法。以第i天为分界线,计算第i天之前进行一次交易的最大收益preProfit[i],和第i天之后进行一次交易的最大收益postProfit[i],最后遍历一遍,max{preProfit[i] + postProfit[i]} (0≤i≤n-1)就是最大收益。第i天之前和第i天之后进行一次的最大收益求法同Best Time to Buy and Sell Stock I。
代码:时间O(n),空间O(n)。
解决方案一:
比较巧妙的方案
- class Solution {
- public int maxProfit(int[] prices) {
-
- int n=prices.length;
-
- if(prices.length<2)return 0;
-
- //第i天之前的最大利益
- int[] preProfit = new int[n];
- //第i天之后的最大利益
- int[] postProfit = new int[n];
- //总的最大利润
- int max = Integer.MIN_VALUE;
-
- //如果今天卖出之后,当前的最大利益小于之前的,则不卖出。否则今天卖出,最大利益得到更新
- int minBuy = prices[0];
- for(int i = 1;i<n;i++){
- minBuy = Math.min(minBuy, prices[i]);
- preProfit[i] = Math.max(preProfit[i-1], prices[i] - minBuy);
- }
- //如果最大价格减掉今天价格比明天以后买入的最大收益大,就用最大价格减掉今天价格,否则,用明天以后买入的最大收益
- //采用倒序,如果在今天买入,在今后能卖出的最大收益为postProfit小于之后买入的,则今天不买入,否则在今天买入,最大收益值得到更新。再继续和前一天比较
- int maxSell = prices[n-1];
- for(int i = n-2;i>=0;i--){
- maxSell = Math.max(maxSell, prices[i]);
- postProfit[i] = Math.max(postProfit[i+1], maxSell-prices[i]);
- }
- //求出两次交易的和,与总的最大利润进行比较
- for(int i = 0;i<n;i++){
- max = Math.max(preProfit[i] + postProfit[i], max);
- }
- return max;
-
- }
- }
下面参考了leetcode上的比较优秀的解答方案
解决方案二:
- class Solution {
- public int maxProfit(int[] prices) {
- if(prices == null || prices.length < 2) {
- return 0;
- }
-
- int hold1 = 0 - prices[0];
- int notHold1 = 0;
- int hold2 = notHold1 - prices[0];
- int notHold2 = 0;
-
- for(int i = 1; i < prices.length; i++) {
- notHold2 = Math.max(notHold2, hold2 + prices[i]);
- hold2 = Math.max(hold2, notHold1 - prices[i]);
- notHold1 = Math.max(notHold1, hold1 + prices[i]);
- hold1 = Math.max(hold1, 0 - prices[i]);
- }
-
- return notHold2;
- }
- }
解决方案三:
该方案在测试集里运行时间为2ms
- class Solution {
- public int maxProfit(int[] prices) {
-
- int buy1 = Integer.MAX_VALUE;
- int sell1 = 0;
- int buy2 = Integer.MAX_VALUE;
- int sell2 = 0;
-
- for(int p : prices) {
-
- buy1 = Math.min(buy1, p);
- sell1 = Math.max(sell1, p - buy1);
- buy2 = Math.min(buy2, p - sell1);
- sell2 = Math.max(sell2, p-buy2);
- }
-
- return sell2;
- }
- }
参考来源:
https://blog.csdn.net/u014609111/article/details/53508905
https://blog.csdn.net/ChenVast/article/details/78950392
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。