1、题目描述
给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
2、示例
输入: [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
3、题解
基本思想:动态规划,dp[i][0 or 1 or 2]表示第i天手上是否持有股票或冷冻期状态下的最大收益
- dp[i][0]表示第i天手中不持有股票状态下的最大收益;dp[i][1]表示第i天手中持有股票状态下的最大收益;dp[i][2]表示第i天冷冻期状态下的最大收益
- 初始化第0天状态值
- 遍历第1天之后的每天股票情况
- 第i天手中没有股票下的最大收益等于max(1、前一天没有股票状态下的收益;2、前一天冷冻期状态下的收益)
- 第i天手中有股票下的最大收益等于max(1、买入股票:前一天的没有股票下的最大收益-今天股票价格;2、无操作:前一天有股票状态下的收益)
- 第i天冷冻期下的收益等于前一天有股票状态今天卖出股票下的收益,也就是前一天的持有股票下的收益+今天股票价格
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
int maxProfit(vector<int>& prices) {
//基本思想:动态规划,dp[i][0 or 1 or 2]表示第i天手上是否持有股票或冷冻期状态下的最大收益
//dp[i][0]表示第i天手中不持有股票状态下的最大收益
//dp[i][1]表示第i天手中持有股票状态下的最大收益
//dp[i][2]表示第i天冷冻期状态下的最大收益
if(prices.size()<2) return 0;
//dp[i][0 or 1 or 2]表示第i天,手上是否持有股票或冷冻期状态下的最大收益
vector<vector<int>> dp(prices.size(), vector<int>(3, 0));
//初始化第0天状态值
dp[0][0]=0;
dp[0][1]=-prices[0];
dp[0][2]=0;
//遍历第1天之后的每天股票情况
for(int i=1;i<prices.size();i++)
{
//第i天手中没有股票下的最大收益等于max(1、前一天没有股票状态下的收益;2、前一天冷冻期状态下的收益)
dp[i][0] = max(dp[i-1][0], dp[i-1][2]);
//第i天手中有股票下的最大收益等于max(1、买入股票:前一天的没有股票下的最大收益-今天股票价格;2、无操作:前一天有股票状态下的收益)
dp[i][1] = max(dp[i-1][0]-prices[i], dp[i-1][1]);
//第i天冷冻期下的收益等于前一天有股票状态今天卖出股票下的收益,也就是前一天的持有股票下的收益+今天股票价格
dp[i][2] = dp[i-1][1]+prices[i];
}
//最后最大收益一定是手中不持有股票,所以返回最后一天不持有股票状态下和冷冻期状态下中的最大值
return max(dp[prices.size()-1][0],dp[prices.size()-1][2]);
}
};
int main()
{
vector<int> prices={1,2,3,0,2};
Solution solute;
cout<<solute.maxProfit(prices)<<endl;
return 0;
}