LeetCode-【动态规划】-最佳买卖股票时机含冷冻期

版权声明:转载注明出处就可以了。。。 https://blog.csdn.net/zw159357/article/details/82594607

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

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

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

示例:

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

题解:对任意一天共有两种状态存在,一是持有股票,二是不持有股票(包括冷冻期)。

持有股票:

1.今天买入 2.前一天没有卖出 。最终最大利润去这两中情况的最大值。

未持有股票

1.和两天前一样未持有(包括冷冻期)2.今天卖出。最终最大利润去这两中情况的最大值。

根据以上分析,分别建立持有股票数组h[n],dh[n],其中n是天数,那么有

h[i]=max(h[i-1],dh[i-2]-prices[i]);

dh[i]=max(dh[i-1],h[i-1]+prices[i]);

class Solution {
    public int maxProfit(int[] prices) {
        int n=prices.length;
        if(n==0)
            return 0;
        int[] dh=new int[n];
        int[] h=new int[n];
        dh[0]=0;
        h[0]=-prices[0];
        for(int i=1;i<n;i++){
            dh[i]=Math.max(dh[i-1],h[i-1]+prices[i]);
            if(i>=2)
                h[i]=Math.max(h[i-1],dh[i-2]-prices[i]);
            else
                h[i]=Math.max(h[i-1],-prices[i]);
        }
        return dh[n-1];
    }
}

 

猜你喜欢

转载自blog.csdn.net/zw159357/article/details/82594607