题目描述:
给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
示例:
输入: [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
方法1:动态规划
主要思路:
(1)对于当天所可以获得利润和三个状态有关:
(a)当前持有股票(可能是一直持有的,也可能是刚买的);
(b)当前处于冷冻状态,则一定是前一天刚卖掉股票;
(c)当前不持有股票(可能是一直不持有,也可能是刚从冷冻状态转移而来);
(2)根据上述解释,设定动态数组vector<vector> dp(size_prices,vector(3,0));则:
(a)dp[ i ][ 0 ]表示当前持有股票下可以获得最大利润,则根据两个可能的转化方式,则 dp[i][0]=max(dp[i-1][0],dp[i-1][2]-prices[i]);
(b)dp[ i ][ 1 ]表示当前处于冷冻状态下可获得最大的利润,则一定是从持有股票获得最大利润下,直接卖出获得唯一的转化方式,既 dp[i][1]=dp[i-1][0]+prices[i];
(c)dp[ i ][ 2 ]表示当前不持有股票下可获得最大的利润,则可能是一直没有持有,或刚从冷冻状态下转化过了,既dp[i][2]=max(dp[i-1][2],dp[i-1][1]);
(3)最后返回冷冻和卖出两种状态下的最大值即可;
class Solution {
public:
int maxProfit(vector<int>& prices) {
if(prices.size()<2)
return 0;
int size_prices=prices.size();
vector<vector<int>> dp(size_prices,vector<int>(3,0));
dp[0][0]=-prices[0];//初始化,既第一天买入,因为此状态下对应一定持有
for(int i=1;i<size_prices;++i){
dp[i][0]=max(dp[i-1][0],dp[i-1][2]-prices[i]);//持有状态下可以获得最大的利润
dp[i][1]=dp[i-1][0]+prices[i];//冷冻状态下可以获得最大的利润
dp[i][2]=max(dp[i-1][2],dp[i-1][1]);//卖出状态下可以获得最大的利润
}
return max(dp[size_prices-1][1],dp[size_prices-1][2]);//返回最大值
}
};
方法2:动态规划,压缩使用内存
主要思路:
(1)观察方法1,可知当天的三种可以获得利润的方式,只和前一天的状态有关,故可以考虑将数组压缩,只保留前一天的状态即可;
class Solution {
public:
int maxProfit(vector<int>& prices) {
if(prices.size()<2){
return 0;
}
int size_prices=prices.size();
//用于保存前一天的状态
int profit0=-prices[0];
int profit1=0;
int profit2=0;
for(int i=1;i<size_prices;++i){
int new_profit0=max(profit0,profit2-prices[i]);
int new_profit1=new_profit0+prices[i];
int new_profit2=max(profit1,profit2);
//更前前一天的状态
profit0=new_profit0;
profit1=new_profit1;
profit2=new_profit2;
}
return max(profit1,profit2);
}
};