股票问题3。给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
一次遍历法。
时间紧,请直接看代码。
时间复杂度: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
}
}
执行结果如下: