给定一个整数数组prices,其中第 prices[i] 表示第 i 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
输入: prices = [1,2,3,0,2] 输出: 3
思路:
将买卖状态分为三个部分:
目前持有一支股票,对应的「累计最大收益」记为 pro[i][0];
目前不持有股票,并且处于冷冻期中,对应的「累计最大收益」记为 pro[i][1];
(处于冰冻期指当天卖出之后,后一天处于冰冻期。eg:pro[i][1]代表第i天卖出,第i天结束后是冰冻期)
目前不持有股票,并且不处于冷冻期中,对应的「累计最大收益」记为 pro[i][2]。
持有股票时:一种是昨天就有的股票,现收益仍为pro[i - 1][0];另一种是今天新买入的股票,利益为上一天的利益(不处于冰冻期)减去今天的花费,pro[i - 1][2] - prices[i]。
不持有股票且处于冰冻期时:表示当天卖出了股票,pro[i - 1][0] + prices[i]。
不持有股票且不处于冰冻期时:一种是昨天也未处于冰冻期,现收益仍为pro[i - 1][2];另一种是昨天处于冰冻期,pro[i - 1][1]。
最终答案为:max(pro[n - 1][1],pro[n - 1][2])。
代码:
class Solution {
public int maxProfit(int[] prices) {
int n = prices.length;
if(n == 0) return 0;
int[][] pro = new int[n][3];
pro[0][0] = -prices[0];
for(int i = 1;i < n; i ++) {
pro[i][0] = Math.max(pro[i - 1][0], pro[i - 1][2] - prices[i]);
pro[i][1] = pro[i - 1][0] + prices[i];
pro[i][2] = Math.max(pro[i - 1][1], pro[i - 1][2]);
}
return Math.max(pro[n - 1][1], pro[n - 1][2]);
}
}