昨天题目是买卖两次,这里是设定好k次
2次的时候一共包含了5中情况 那么k次就包含了2 k + 1种情况
初始化只有1 , 3 ,5 也就是买入的状况是有值的,所以初始化时候只需要看这些情况知道2 k
之后只要把之前两次的做一下抽象就可以了
这道题感觉稍微麻烦一点
首先包含冷冻期其中有4种情况
1.持有股票的情况
2.不持有股票情况(之前就是冷冻期了,但是之后一直没买)
3.不持有股票(今天刚刚卖出股票)
4.昨天刚刚卖出股票,今天就是冷冻期
所以对应的递推公式有
达到买入股票状态(状态一)即:dp[i][0],有两个具体操作:
- 操作一:前一天就是持有股票状态(状态一),dp[i][0] = dp[i - 1][0]
- 操作二:今天买入了,有两种情况
- 前一天是冷冻期(状态四),dp[i - 1][3] - prices[i]
- 前一天是保持卖出股票的状态(状态二),dp[i - 1][1] - prices[i]
那么dp[i][0] = max(dp[i - 1][0], dp[i - 1][3] - prices[i], dp[i - 1][1] - prices[i]);
达到保持卖出股票状态(状态二)即:dp[i][1],有两个具体操作:
- 操作一:前一天就是状态二
- 操作二:前一天是冷冻期(状态四)
dp[i][1] = max(dp[i - 1][1], dp[i - 1][3]);
达到今天就卖出股票状态(状态三),即:dp[i][2] ,只有一个操作:
昨天一定是持有股票状态(状态一),今天卖出
即:dp[i][2] = dp[i - 1][0] + prices[i];
达到冷冻期状态(状态四),即:dp[i][3],只有一个操作:
昨天卖出了股票(状态三)
dp[i][3] = dp[i - 1][2];
很简单 和2是一样的 只是要多加一个手续费