[代码随想录]Day42-动态规划part10

发布时间 2023-09-12 10:11:27作者: WtcSky

题目:121. 买卖股票的最佳时机

思路:

贪心做起来更简单;dp多此一举……状态0是有买入,状态1是

代码:

func maxProfit(prices []int) int {
    lens := len(prices)
    if lens == 0 {
        return 0
    }
    dp := make([][]int, lens)
    for i := 0; i < lens; i++ {
        dp[i] = make([]int, 2)
    }
    dp[0][0] = -prices[0]
    dp[0][1] = 0
    for i:=1; i < lens;i++ {
        dp[i][0] = max(dp[i-1][0],-prices[i])
        dp[i][1] = max(dp[i-1][1], dp[i-1][0] + prices[i])
    }
    return dp[lens-1][1]
}
func max(a,b int)int{
    if a > b{
        return a
    }
    return b
}

参考:

代码随想录

题目:122. 买卖股票的最佳时机 II

思路:

也是贪心简单,具体参考Day28的贪心
dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] - prices[i])今天持有的状态取决于:

  1. 昨天持有
  2. 昨天不持有,今天买
    dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i])今天不持有的状态取决于:
  3. 昨天不持有
  4. 昨天持有,今天卖了

代码:

func maxProfit(prices []int) int {
    dp := make([][]int, len(prices))
    status := make([]int, len(prices) * 2)
    for i := range dp {
        dp[i] = status[:2]
        status = status[2:]
    }
    dp[0][0] = -prices[0]
    
    for i := 1; i < len(prices); i++ {
        dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] - prices[i])
        dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i])
    }
    
    return dp[len(prices) - 1][1]
}
func max(a,b int ) int {
    if a > b {
        return a
    }
    return b
}

参考:

代码随想录