题目描述
给定一个整数数组prices,其中第 prices[i] 表示第 i 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
代码
/*
dp含义:
dp[i][0]:表示持有股票的状态
dp[i][1]:不持有股票,本日卖出,下一天为冷冻期
dp[i][2]:不持有股票,本日无卖出,下一天非冷冻期
递推公式:
第i天时,我们可以在不违反规则的前提下进行买入和卖出操作。此时第i天的状态会从第i-1天状态转移而来
我们可以不进行任何操作,此时第i天的状态就等同于第i-1天的状态。
对于dp[i][0]:对应两种状态。
第i-1天就已经持有->dp[i-1][0]
第i天买入,那么第i-1天就不能持有股票并且不处于冷冻期->dp[i-1][2]
dp[i][0] = max(dp[i-1][0],dp[i-1][2]-prices[i])
对于dp[i][1]: 本日卖出 : (条件:昨天持有)
dp[i][1] = dp[i-1][0]+prices[i]
对于dp[i][2]: 不持有,但也不是卖出 :
1.昨天卖出,不持有->dp[i-1][1]
2.昨天没卖出,但也不持有->dp[i-1][2]
dp[i][2] = max(dp[i-1][1],dp[i-1][2])
初始化:
dp[0][0] = -prices[0]
dp[0][1] = 0
dp[0][2] = 0
便利顺序:
for(int i=1;i<prices.size();i++)
*/
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(), vector<int>(3, 0));
dp[0][0] = -prices[0];
dp[0][1] = 0;
dp[0][2] = 0;
for (int i = 1; i < prices.size(); i++) {
// 持有股票: 1.昨天持有股票 2.本日买入(条件:昨天不持有,且不是卖出)
dp[i][0] = max(dp[i - 1][0], dp[i - 1][2] - prices[i]);
//本日卖出 : (条件:昨天持有)
dp[i][1] = dp[i - 1][0] + prices[i];
// 不持有,但也不是卖出 : 1.昨天卖出,不持有 2.昨天没卖出,但也不持有
dp[i][2] = max(dp[i - 1][1], dp[i - 1][2]);
}
return max(dp[prices.size() - 1][1], dp[prices.size() - 1][2]);
}
};