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 } }
执行结果如下: