2021-07-08：股票问题5。给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。​设计一个算法计算出最大利润

package main

import "fmt"

func main() {
    arr := []int{1, 2, 3, 4, 1, 6, 7}
    ret := maxProfit(arr)
    fmt.Println(ret)
}

func maxProfit(prices []int) int {
    if len(prices) < 2 {
        return 0
    }
    buy1 := getMax(-prices[0], -prices[1])
    sell1 := getMax(0, prices[1]-prices[0])
    sell2 := 0
    for i := 2; i < len(prices); i++ {
        tmp := sell1
        sell1 = getMax(sell1, buy1+prices[i])
        buy1 = getMax(buy1, sell2-prices[i])
        sell2 = tmp
    }
    return sell1
}

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