代码随想录day50【动态规划】买卖股票的最佳时机III

买卖股票的最佳时机III

力扣题目链接
与前题的区别：
至多买卖两次，这意味着可以买卖一次，可以买卖两次，也可以不买卖。

1. dp数组定义：
   dp[i][0] 不操作（其实可以不用，因为没用到）
   dp[i][1]第一次持有
   dp[i][2]第一次不持有
   dp[i][3]第二次持有
   dp[i][4]第二次不持有

2. 递推公式
   dp[i][0]=dp[i-1][0]
   dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i])
   dp[i][2]=max(dp[i-1][2],dp[i-1][1]+prices[i])
   dp[i][3]=max(dp[i-1][3],dp[i-1][2]-prices[i])//延续前一天的状态（持有），或第一次不持有的状态，买入）注意这个
   dp[i][4]=max(dp[i-1][4],dp[i-1][3]+prices[i])//延续前一天的状态（持有），或第二次持有的状态，卖出）注意这个
   初始化
   dp[0][0]=0
   dp[0][1]=-prices[0]
   dp[0][2]=0
   dp[0][3]=-prices[0]
   dp[0][4]=0

var maxProfit = function(prices) {
    let dp=new Array(prices.length).fill(0).map(ele=> new Array(5).fill(0))
    dp[0][0]=0
    dp[0][1]=-prices[0]
    dp[0][2]=0
    dp[0][3]=-prices[0]
    dp[0][4]=0

    for(let i=1;i<prices.length;i++){
        dp[i][0]=dp[i-1][0]
        dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i])
        dp[i][2]=Math.max(dp[i-1][2],dp[i-1][1]+prices[i])
        dp[i][3]=Math.max(dp[i-1][3],dp[i-1][2]-prices[i])//延续前一天的状态（持有），或第一次不持有的状态，买入）注意这个
        dp[i][4]=Math.max(dp[i-1][4],dp[i-1][3]+prices[i])//延续前一天的状态（持有），或第二次持有的状态，卖出）注意这个
    }
    return dp[prices.length-1][4]
};

买卖股票的最佳时机IV

力扣题目链接
与前题的区别：
至多买卖k次，
递推公式改为：用变量表示。

var maxProfit = function(k, prices) {
    let dp=new Array(prices.length).fill(0).map(ele=> new Array(2*k+1).fill(0))
    for(let j=1;j<2*k;j+=2){//奇数为负
            dp[0][j]=-prices[0]
        }
    for(let i=1;i<prices.length;i++){
        for(let j=0;j<2*k;j+=2){
             dp[i][j+1]=Math.max(dp[i-1][j+1],dp[i-1][j]-prices[i])
            dp[i][j+2]=Math.max(dp[i-1][j+2],dp[i-1][j+1]+prices[i])
        }
       
    }
    return dp[prices.length-1][2*k]
}