题解 | #股票交易的最大收益（二）#

package main

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 * 两次交易所能获得的最大收益
 * @param prices int整型一维数组 股票每一天的价格
 * @return int整型
*/
func maxProfit( prices []int ) int {
    // write code here
    l := len(prices)
    if l == 0 || l == 1 {
        return 0
    }

    dp := make([][]int, l)
    for i:=0; i<len(dp); i++{
        dp[i] = make([]int, 5)
    }

    dp[0][1] = -prices[0]
    dp[0][3] = -prices[0]

    for i:=1; i< len(prices); i++{
        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])
    }

    return dp[len(prices)-1][4]
}

func max(a, b int) int{
    if a>b{return a}
    return b
}