【Java力扣算法】LeetCode 309 Best Time to Buy and Sell Stock with Cooldown(动态规划)

题目:

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

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

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

示例:

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

 

思路:

      动态规划就是通过状态转移公式,也就是   当前状态和 下一状态的对比,比出哪个更好就取用哪个。

      对于这个题来说就是  sell的时候,其实就是比较  昨天卖比较好还是今天卖比较好,所以就是sell[i] = max(sell[i] , sell[i-1]),因为sell[i] =sell[i]就没意义了,所以这时候就要找一个sell[i]的替代品,所以推理一下第i天卖sell[i]就=当前价格price[i] - 从前一天算起的最合理买入价格buy[i-1]。由此可以推出一个状态转移方程     sell(i) = max(sell(i - 1) ,price(i) - buy(i - 1))。

      同理  买入的时候就是比较今天买比较好还是昨天买比较好。buy[i] = max(buy[i] , buy[i - 1])--->max(buy[i - 1] , sell(i - 2) - price(i))。其中替换buy(i) 的sell(i - 2) - price(i) 的意思就是上一次卖出后 又买入挣得钱。为什么是sell(i -2)原因就是有一天的冻结期,如果没有冻结期的话就是sell(i - 1)了。

      这时候再把 冻结期加上,因为冻结期就是因为前一天卖了,所以第二天才会冻结,所以cooldown(i) = sell(i - 1)。

      所以先定义dp二维数组,因为有三个变量,所以用第二维用来区分(dp[i][0] 表示卖的时候可以挣的最多的钱,dp[i][1]表示买的时候可以挣的最多的钱,dp[i][2]表示冻结时期时候  能挣的最多的钱)

通过一个例子来逐步解释解释:

价格  1  2  3  0  1  0  6
冻结  0  0  1  2  2  2  2
卖出  0  1  2  2  2  2  8  比较前一天卖整的多还是当天买挣的多sell(i-1),sell(i)--> sell(i-1) buy(i-1)+price(i)
买入 -1 -1 -1  1  1  2  2  比较前一天买挣得多还是当天买挣得多buy(i-1),buy(i)  --> buy(i-1) sell(i-2)-price(i) 

 

 

代码:

    public static int maxProfit(int[] prices) {
        int[][] dp=new int[prices.length][3];
        //添加边界判断条件
        if(prices == null || prices.length < 2)  return 0;
        //定义三个数组,买的 卖的 冻结的时刻可以挣的最多的钱。
        dp[0][0]=0;//sell
        dp[0][1]= - prices[0];//buy
        dp[0][2]=0;//cooldown
        for(int i = 1 ;i < prices.length ;i++){
            dp[i][0]=Math.max(dp[i - 1][0],dp[i - 1][1] + prices[i]);
            dp[i][1]=Math.max(dp[i - 1][1],dp[i - 1][2] - prices[i]);
            dp[i][2]=dp[i - 1][0];
        }
        return Math.max(dp[prices.length - 1][0],dp[prices.length - 1][2]);
    }

 

结果:

【Java力扣算法】LeetCode 309 Best Time to Buy and Sell Stock with Cooldown(动态规划)_第1张图片

 

你可能感兴趣的:(java算法习题程序)