算法训练营Day51(买卖股票总结）

 309.最佳买卖股票时机含冷冻期  

309. 买卖股票的最佳时机含冷冻期 - 力扣（LeetCode）

dp数组：

        模拟状态：买买买 卖  冻卖卖卖  买  卖

        dp[i][0] :持有股票

        dp[i][1]: 不持有股票，但是之前卖出去的

        dp[i][2]: 不持有股票，是刚卖出去的

        dp[i][3]:冷冻期

递推公式：

        dp[i][0] :

                前一天本来就持有dp[i-1][0]

                刚买的：        

                        冷冻期刚过买的：dp[i-1][3]-prices[i]

                        卖出去好几天了才买的：dp[i-1][1]-prices[i]

        dp[i][1] :

                前一天就保持卖出：dp[i-1][1]

                刚过完冷冻期：dp[i-1][3]

        dp[i][2]:

                昨天是买入的，刚卖出 dp[i-1][0]+prices[i]

        dp[i][3]:

                昨天刚卖出 dp[i-1][2]

      初始化：

      dp[0][0]  = -prices[0]          

        dp[0][1] =0;//当天卖当天买满足递推公式合理

        dp[0][2] =0;

        dp[0][3] = 0;

class Solution {
    public int maxProfit(int[] prices) {
        int [][] dp = new int[prices.length][4];
        dp[0][0] = -prices[0];

        for(int i = 1;i<prices.length;i++){
            dp[i][0]=Math.max(dp[i-1][0],Math.max(dp[i-1][3]-prices[i],dp[i-1][1]-prices[i]));
            dp[i][1]=Math.max(dp[i-1][1],dp[i-1][3]);
            dp[i][2]=dp[i-1][0]+prices[i];
            dp[i][3]=dp[i-1][2];
        }
        return Math.max(dp[prices.length-1][3],Math.max(dp[prices.length-1][1],dp[prices.length-1][2]));
    }
}

 714.买卖股票的最佳时机含手续费  

714. 买卖股票的最佳时机含手续费 - 力扣（LeetCode）

卖的时候-费用就好了，秒了

class Solution {
    public int maxProfit(int[] prices, int fee) {
        //0持有 1不持有
        int [][] dp = new int[prices.length][2];
        dp[0][0] = -prices[0];
        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][0]+prices[i]-fee);
        }
        return dp[prices.length-1][1];
    }
}

总结

买卖股票理解好买卖股票状态和逻辑关系就可以了，并没想象中的复杂，按照动态规划5步曲就很好理解。