LeetCode(309):最佳买卖股票时机含冷冻期
题目描述
给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。int[ ] prices 股票价格。
解题思路
1、动态规划
与分治算法思想一致,都是将复杂问题分解成若干小问题,最后通过小问题进行求。不同的是,分治算法各个小问题之间是相互独立的;而动态规划算法中各个小问题之间相互影响。
2、题解
定义:若买入股票则总金额-prices[i];若卖出股票则总金额+prices[i]。
第i天可以处于三种状态中的一种:
0:第i天持有股票。
1:第i天卖出股票,即第i+1天不能购买股票。
2:第i天未持有股票。
创建数组profit[prices.length][3],用来存储某天处于某种状态下的最大盈利。
若第i天处于0:代表目前持有股票,则profit[i][0] = Math.max(profit[i-1][0],profit[i-1][2]-prices[i]);
若第i天处于1:代表今天抛出股票,则profitp[i][1]=profit[i-1][0]+prices[i];
若第i天处于2:代表今天未持有股票,则profit[i][2]=Math.max(profit[i-1][1],profit[i-1][2]);
初始值:profit[0][0] = - prices[0];profit[0][1]=0;profit[0][2]=0;
代码实现
public int maxProfit(int[] prices) {
//考虑特殊情况
if(prices.length < 2){
return 0;
}
//创建数组存放对应状态下的最大盈利
int[][] profit = new int[prices.length][3];
//初始化
profit[0][0] = -prices[0];
profit[0][1] = 0;
profit[0][2] = 0;
//循环从第1天开始
for(int i = 1;i < prices.length;i++){
profit[i][0] = Math.max(profit[i - 1][0],profit[i-1][2]-prices[i]);
profit[i][1] = profit[i - 1][0] + prices[i];
profit[i][2] = Math.max(profit[i-1][2],profit[i-1][1]);
}
//返回最终结果,最后一天不能有剩余股票未抛出
return Math.max(profit[prices.length - 1][1],profit[prices.length - 1][2]);
}