文章目录
29.买卖股票的最佳时机含冷冻期
29.1题目
给定一个整数数组prices
,其中第 prices[i]
表示第 *i*
天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
- 卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
- 示例一:
输入: prices = [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
- 示例二:
输入: prices = [1]
输出: 0
29.2解法
29.2.1动规思路
-
确定dp数组以及下标含义:dp(i)(股票状态)
i为天数、股票状态有四种:
- 状态一:持有股票(今天买入、之前就买入了)
- 状态二:保持卖出股票状态(两天前卖出股票,度过一天冷冻期)
- 状态三:今天卖出股票
- 状态四:今天为冷冻期
-
确定递归公式;
- 持有股票状态:
- 情况一:前一天就是持有股票状态
- 情况二:今天买入,前一天是冷冻期
- 情况三:今天买入,前一天是保持卖出股票状态
- dp(i)(0)=max(dp(i-1)(0),max(dp(i-1)(3)-prices[i],dp(i-1)(1)-prices[i])
- 卖出股票状态:
- 情况一:前一天就是保持卖出股票状态
- 情况二:前一天是冷冻期
- dp(i)(1)=max(dp(i-1)(1),dp(i-1)(2))
- 今天卖出股票状态:
- dp(i)(2)=dp(i-1)(0)+prices[i]
- 冷冻期:前一天卖出
- dp(i)(3)=dp(i-1)(2)
- 持有股票状态:
-
dp数组初始化:dp(0)(0)=-prices[0]
-
确定遍历顺序:从前往后
29.2.2代码实现
public int maxProfit(int[] prices) {
int len=prices.length;
int[][] dp=new int[len][4];
dp[0][0]=-prices[0];
for(