Leetcode 309.买卖股票的最佳时期含冷冻期（你也想成为股票高手吗o〃ω〃o，718）

题目

26年7月18日（首刷自解）

Go 复制代码

func maxProfit(prices []int) int {
    //dp数组含义.dp[i][j]表示在第i天，状态为j时的最多钱
    dp:=make([][]int,len(prices)+1)
    for i,_ :=range dp{
        dp[i]=make([]int,3)
    }
    dp[0][0]=0
    dp[0][1]=-prices[0]
    dp[0][2]=0
    dp[1][0]=0
    dp[1][1]=-prices[0]
    dp[1][2]=0
    for i:=2;i<=len(prices);i++{
        dp[i][0]=maxplus(dp[i-1][1]+prices[i-1],dp[i-1][0],dp[i-1][2])
        dp[i][1]=max(dp[i-2][0]-prices[i-1],dp[i-1][1])
        dp[i][2]=dp[i-1][0]
    }
    return max(dp[len(prices)][0],dp[len(prices)][2])
}
func maxplus(a,b,c int) int{
    res1:=0
    if a>=b{
        res1=a
    }else{
        res1=b
    }
    if res1>=c{
        return res1
    }else{
        return c
    }
}

思路

这道题和上一道题都是自己做出来的，属于举一反三了，这道题现在给你一个冷冻期（术士熔断），卖股票后要等一天（开领域），其实能想到，无非多了一种状态，所以一共有三种状态，持有，没持有，冷冻期

这里我最开始的错误代码是这样的

Go 复制代码

func maxProfit(prices []int) int {
    //dp数组含义.dp[i][j]表示在第i天，状态为j时的最多钱
    dp:=make([][]int,len(prices)+1)
    for i,_ :=range dp{
        dp[i]=make([]int,3)
    }
    dp[0][0]=0
    dp[0][1]=-prices[0]
    dp[0][2]=0
    dp[1][0]=0
    dp[1][1]=-prices[0]
    dp[1][2]=0
    for i:=2;i<=len(prices);i++{
        dp[i][0]=maxplus(dp[i-1][1]+prices[i-1],dp[i-1][0],dp[i-1][2])
        dp[i][1]=max(dp[i-1][0]-prices[i-1],dp[i-1][1])
        dp[i][2]=dp[i-1][0]
    }
    return max(dp[len(prices)][0],dp[len(prices)][2])
}
func maxplus(a,b,c int) int{
    res1:=0
    if a>=b{
        res1=a
    }else{
        res1=b
    }
    if res1>=c{
        return res1
    }else{
        return c
    }

在我手动模拟后发现问题

可以看到我dp $4$ $1$ 其实是出问题了，看看dp $4$ $1$ 如果算出来等于2，意味着第一天买入，第三天卖出，而我第四天是持有状态，说明我第四天又买入了，但是熔断期还没过怎么可能买？

所以递推公式dp $i$ $1$ 其实应该为max(dp[i-2][0]-prices[i-1],dp[i-1][1])，是dp $i-2$ $0$

其实也好理解，如果你今天持有的状态是因为今天买入而不是之前就持有，说明你在之前某个点是卖出了股票的，这个点不可能是昨天，因为如果是昨天那么今天会是熔断期，所以最早也得是前天，

注意相应的初始化，也要改，i=0和i=1都需要初始化，然后循环从i=2开始