Leetcode 309. Best Time to Buy and Sell Stock with Cooldown

最新推荐文章于 2020-02-22 21:21:18 发布

原创最新推荐文章于 2020-02-22 21:21:18 发布 · 157 阅读

0 ·

CC 4.0 BY-SA版权

DP 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种使用动态规划解决股票买卖问题的方法，通过定义持有和未持有股票两种状态，实现寻找股票买卖的最佳时机以获得最大利润。代码示例清晰展示了状态转移的过程。

Problem:

在这里插入图片描述

Analysis:

DP is like induction prove.

define the base case
define the inductive rule

Code:

public int maxProfit(int[] prices) {
        
        if (prices == null || prices.length <= 1) return 0;
        
        int n = prices.length;
        int[] hold = new int[n];
        int[] unhold = new int[n];
        
        hold[0] = - prices[0];
        hold[1] = Math.max(-prices[0], -prices[1]);
        
        unhold[0] = 0;
        unhold[1] = Math.max(hold[0]+prices[1], 0);
        
        for (int i=2; i<n; i++){
            
            hold[i] = Math.max(unhold[i-2] - prices[i], hold[i-1]);
            unhold[i] = Math.max(hold[i-1] + prices[i], unhold[i-1]);      
            
        }
        
        return unhold[n-1];
        
    }