动态规划中等 leetcode309. 最佳买卖股票时机含冷冻期

本文主要参考了“liweiwei1419”的解答:
https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown/solution/dong-tai-gui-hua-by-liweiwei1419-5/

import java.lang.Math;
class Solution {
    /*
   
    dp[i][j] i表示第几天,j表示状态:持股,未持股,冷冻。 
    dp[i][j]表示从第0天开始到第i天j状态下的最大收益。
    动态转移方程根据状态分为三种情况:
    dp[i][0]是冷冻状态收益(本来冷冻状态只有一天,
    为了方便把卖掉股票后到再次买股票前都称之为冷冻,与为持股状态有重叠,但不影响)
    其等于上一日的未持股的收益。
    dp[i][1]是持股状态收益(分为买股和前几天买股以后一直未抛售),
    其等于Math.max(昨天是冷冻减去今天的股价,昨天的持股收益)
    dp[i][2]是未持股状态收益(分为今天刚卖了股和前几天就卖了一直还没买新股)
    其等于Math.max(昨天持股的收益加上今天的股价,昨天的未持股收益)

    最终结果是想求最后一天的收益,且状态一定是冷冻或者是未持股。
    (因为最后一天买股或者没掉之前的股都会减少收益)
    即就是Math.max(dp[n-1][0],dp[n-1][2])
    从1开始遍历就可得到最后结论。


    总结:最大收益就是第n-1天冷冻状态或者未持股状态下的收益。
    通过动态转移方程得知,只要知道第n-2天的三者状态的收益就可知道第n-1天的收益。
    而第n-2天的信息可由第n-3天的得知,所以只要从第0天开始遍历就一定得知最终结果。
    */
    public int maxProfit(int[] prices) {
        int n=prices.length;
        if(n<2){
            return 0;
        }
        int [][]dp=new int[n][3];
        dp[0][0]=0;//basecase设初值,当是第0天未持股,冷冻都是没收益的,因为只有卖掉股票才会有收益
        dp[0][1]=-prices[0];//第0天持股的收益是负的,为第0天的股价的负数。
        dp[0][2]=0;
        for(int i=1;i

你可能感兴趣的:(#,动态规划,leetcode做题笔记)