【动态规划】 309 最佳买卖股票含冷冻期

题目

给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）: 你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。 卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。 输入: [1,2,3,0,2] 输出: 3 解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]

思路

股票问题不含冷冻期 dp【】【3】增加一个维度，是冷冻期的情况。冷冻期前一天只有一种情况，就是现金，也就是卖出 dp表示的是今天的操作带来的收益。 收益是+price，支出是—price

代码

public int maxProfit(int[] prices) { if(prices.length<2) return 0; int [][]dp = new int[prices.length][3]; //0 :现金(卖出) 1：股票（买入） 2:冷冻 dp[0][0] = 0; dp[0][1] = -prices[0]; dp[0][2] = 0; for(int i = 1;i<prices.length;i++){ // 今天是现金：前一天是现金或者前一天买入今天卖出，然后是现金 dp[i][0] = Math.max(dp[i-1][0],dp[i-1][1]+prices[i]); //买入的前一天是冷冻期或者股票 //要买前一天只能是冷冻期或者买入状态 // 今天是股票：前一天是股票或者前一天是冷冻然后今天买入股票 dp[i][1] = Math.max(dp[i-1][2]-prices[i],dp[i-1][1]); //冷冻期前一天是卖出 dp[i][2] = dp[i-1][0]; } return Math.max(dp[prices.length - 1][0], dp[prices.length - 1][2]); }