题目:
给定一个整数数组 prices,其中第 i 个元素代表了第 i 天的股票价格 ;整数 fee 代表了交易股票的手续费用。
你可以无限次地完成交易,但是你每笔交易都需要付手续费。如果你已经购买了一个股票,在卖出它之前你就不能再继续购买股票了。
返回获得利润的最大值。
注意:这里的一笔交易指买入持有并卖出股票的整个过程,每笔交易你只需要为支付一次手续费。
解答:
class Solution:
def maxProfit(self, prices: List[int], fee: int) -> int:
res=0
minprice=prices[0]
n=len(prices)
for i in range(1,n):
if prices[i]<minprice:
minprice=prices[i]
#若利润大于0,则卖出
if prices[i]>minprice+fee:
res+=prices[i]-minprice-fee
minprice=prices[i]-fee #后续价格可能更高,更有利于卖出(难点在此)
return res
方法二:动态规划
class Solution:
def maxProfit(self, prices: List[int], fee: int) -> int:
# dp[i][1]第i天持有的最多现金
# dp[i][0]第i天持有股票所剩的最多现金
n=len(prices)
dp=[[0]*2 for _ in range(n)]
dp[0][0]-=prices[0]
for i in range(1,n):
dp[i][0]=max(dp[i-1][0],dp[i-1][1]-prices[i])
dp[i][1]=max(dp[i-1][1],dp[i-1][0]+prices[i]-fee)
return max(dp[n-1][0],dp[n-1][1])