leetcode【每日一题】309. 最佳买卖股票时机含冷冻期 Java

题干

给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
示例:

输入: [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]

想法

显然动态规划
舍sell[i],buy[i]分别表示i天卖出，买进后 的钱
注意到买有限制，但是卖没有

buy[i]=Math.max(buy[i-1],sell[i-2]-prices[i-1]);
sell[i]=Math.max(sell[i-1],buy[i-1]+prices[i-1]);

其中注意prices[i-1]是指的第i天的价格

对第一个状态方程，
首先第i天我们可能买，可能不买，不买就等于buy[i-1]
如果买，那么意味着昨天一定没卖，于是是拿前天卖了后的钱，减去今天的价格。。

对于i天我们可能卖，卖就是昨天买后的钱再加上今天的价格
不卖，那就一直等于

Java代码

package daily;

public class MaxProfit2 {
    public int maxProfit(int[] prices) {
        int days = prices.length;

        int[] buy = new int[days + 1];
        int[] sell = new int[days + 1];

        if (days == 0 || days == 1) {
            return 0;
        }
        buy[1] = -prices[0];
        sell[0] = 0;
        sell[1] = 0;
        for (int i = 2; i <= days; i++) {
//            其中注意prices[i-1]是指的第i天的价格//            对第一个状态方程，
//            首先第i天我们可能买，可能不买，不买就等于buy[i-1]
//            如果买，那么意味着昨天一定没卖，于是是拿前天卖了后的钱，减去今天的价格。。
//            对于i天我们可能卖，卖就是昨天买后的钱再加上今天的价格
//            不卖，那就一直等于
            buy[i] = Math.max(buy[i - 1], sell[i - 2] - prices[i - 1]);
            sell[i] = Math.max(sell[i - 1], buy[i - 1] + prices[i - 1]);
        }

        return sell[days];

    }

    public static void main(String[] args) {
        MaxProfit2 maxProfit2 = new MaxProfit2();
        int[] test = {1, 2, 3, 0, 2};
        System.out.println(maxProfit2.maxProfit(test));
    }

}

我的leetcode代码都已经上传到我的githttps://github.com/ragezor/leetcode