2021-07-06:股票问题3。给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

福大大 答案2021-07-06:

一次遍历法。
时间紧,请直接看代码。
时间复杂度:O(N)。空间复杂度:O(1)。

代码用golang编写。代码如下:

package main

import "fmt"

func main() {
    arr := []int{1, 1, 10, 1, 1, 9, 3}
    ret := maxProfit(arr)
    fmt.Println(ret)
}

func maxProfit(prices []int) int {
    N := len(prices)
    if N < 2 {
        return 0
    }
    ans := 0
    doneOnceMinusBuyMax := -prices[0]
    doneOnceMax := 0
    min := prices[0]
    for i := 1; i < N; i++ {
        min = getMin(min, prices[i])                                             //最小值
        ans = getMax(ans, doneOnceMinusBuyMax+prices[i])                         //二次交易的最大值
        doneOnceMax = getMax(doneOnceMax, prices[i]-min)                         //一次交易的最大值
        doneOnceMinusBuyMax = getMax(doneOnceMinusBuyMax, doneOnceMax-prices[i]) //一次交易的最大值减去当前值
    }
    return ans
}

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

func getMin(a int, b int) int {
    if a < b {
        return a
    } else {
        return b
    }
}

执行结果如下:
图片


左神java代码