算法训练营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步曲就很好理解。