题目
给定一个数组,它的第 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 。
思路
- 一天结束会处于5种状态:
- 未进行任何操作
- 只进行过一次买操作
- 进行一次买操作和卖操作,即完成了一笔交易
- 完成了一笔交易,进行了第二次买操作
- 完成了两笔交易
- 记录每一种状态的利润(第i-1天)
- 未进行任何操作(利润为0不用记录)
- 只进行过一次买操作(利润为buy1)
- 进行一次买操作和卖操作,即完成了一笔交易(利润为sell1)
- 完成了一笔交易,进行了第二次买操作(利润buy2)
- 完成了两笔交易(利润sell2)
- 记录每一种状态的利润(第i天)
- 不做任何操作或买入一次股票
- 不做任何操作或再买入一次股票的情况卖出股票
- 不做任何操作或在交易一次的情况下再买入股票
- 不做任何操作或再交易完成一次还买入一次股票的情况下再卖出股票
代码
class Solution {
public int maxProfit(int[] prices) {
int n=prices.length;
//-表示买入,+表示卖出
//前一天的状态i-1
//最大利润:buy1,sell1,buy2,sell2
//第一次买入
int buy1=-prices[0];
//完成一次交易(同一天买入并卖出)
int sell1=0;
//完成一次交易,第二次买入(同一天买入并卖出,在买入)
int buy2=-prices[0];
//完成二次买入(同一天买入卖出,再买入卖出)
int sell2=0;
for(int i=1;i<n;i++){
//当天状态i
buy1=Math.max(buy1,-prices[i]);
//sell1为同一天买入卖出,buy1+prices[i]前一天买入后一天卖出
sell1=Math.max(sell1,buy1+prices[i]);
//buy2同一天买入卖出再买入,sell1-prices[i]前一天买入卖出,后一天买入
buy2=Math.max(buy2,sell1-prices[i]);
//sell2同一天完成两次买卖,buy2+prices[i]表示前一天买卖一次,再买入一次,后一天卖出
sell2=Math.max(sell2,buy2+prices[i]);
}
return sell2;
}
}
复杂度
时间复杂度: O(n)
空间复杂度: O(1)