最佳买卖股票含冷冻期
给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
class Solution {
public int maxProfit(int[] prices) {
if(prices == null || prices.length == 0){
return 0;
}
/**
* 三种状态:
* 0 持有股票
* 1 未持有股票且处于冷冻期
* 2 未持有股票且未处于冷冻期
*/
int[][] dp = new int[prices.length][3];
// 买入
dp[0][0] = -prices[0];
for(int i = 1; i < prices.length; i++){
// 第i天持有股票
dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2] - prices[i]);
// 第i天未持有股票且处于冷冻期(当天刚卖)
dp[i][1] = dp[i - 1][0] + prices[i];
// 第i天未持有股票且未处于冷冻期
dp[i][2] = Math.max(dp[i - 1][1], dp[i - 1][2]);
}
// 最后一天肯定不持有股票
return Math.max(dp[prices.length - 1][1], dp[prices.length - 1][2]);
}
}