題目:
假設你有一個數組,其中第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;
}
}