题目
给定一个数组,它的第 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
解题
方法一:动态规划
dp[i][0]
表示无操作的状态
dp[i][1]
表示第一次买入的状态
dp[i][2]
表示第一次卖出的状态
dp[i][3]
表示第二次买入的状态
dp[i][4]
表示第二次卖出的状态
初始化
dp[0][1]=-prices[0];表示第一次买入的收益,
dp[0][3]=-prices[0];表示第二次买入的收益,为什么第一次还没卖出就有第二次买入了呢?因为这样就只需要卖一次就可以得到,第二次卖出。因为题目要求最多买卖2次,这样就可以实现买卖一次。
dp[0][4]初始化为0,表示一次都不买卖,这样初始化可实现一次都不买卖。
于是这样子初始化,可以实现,一次不买卖,买卖一次,买卖两次,之间的比较。选取最大的。
class Solution { public: int maxProfit(vector<int>& prices) { int n=prices.size(); vector<vector<int>> dp(n,vector<int>(5)); dp[0][1]=-prices[0]; dp[0][3]=-prices[0]; for(int i=1;i<n;i++){ dp[i][0]=dp[i-1][0];//这行可以不加 dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i]); dp[i][2]=max(dp[i-1][2],dp[i-1][1]+prices[i]); dp[i][3]=max(dp[i-1][3],dp[i-1][2]-prices[i]); dp[i][4]=max(dp[i-1][4],dp[i-1][3]+prices[i]); } return dp[n-1][4]; } };
之所以要写dp[i][0]=dp[i-1][0]
是为了后面保持统一形式
否则dp[i][1]=max(dp[i-1][1],-prices[i]);