股票问题6。给定一个整数数组 prices,其中第 i 个元素代表了第 i 天的股票价格 ;整数 fee 代表了交易股票的手续费用。你可以无限次地完成交易,但是你每笔交易都需要付手续费。如果你已经购买了一个股票,在卖出它之前你就不能再继续购买股票了。返回获得利润的最大值。注意:这里的一笔交易指买入持有并卖出股票的整个过程,每笔交易你只需要为支付一次手续费。
时间紧。见代码。
时间复杂度:O(N)。空间复杂度:O(1)。
代码用golang编写。代码如下:
package main
import "fmt"
func main() {
arr := []int{1, 2, 3, 4, 1, 6, 7}
fee := 2
ret := maxProfit(arr, fee)
fmt.Println(ret)
}
func maxProfit(arr []int, fee int) int {
if len(arr) < 2 {
return 0
}
N := len(arr)
// 0..0 0 -[0] - fee
bestbuy := -arr[0] - fee
// 0..0 卖 0
bestsell := 0
for i := 1; i < N; i++ {
// 来到i位置了!
// 如果在i必须买 收入 - 批发价 - fee
curbuy := bestsell - arr[i] - fee
// 如果在i必须卖 整体最优(收入 - 良好批发价 - fee)
cursell := bestbuy + arr[i]
bestbuy = getMax(bestbuy, curbuy)
bestsell = getMax(bestsell, cursell)
}
return bestsell
}
func getMax(a int, b int) int {
if a > b {
return a
} else {
return b
}
}
执行结果如下: