题目:买卖股票的最佳时机含手续费
- 给定一个整数数组
prices
,其中第i
个元素代表了第i
天的股票价格 ;非负整数fee
代表了交易股票的手续费用。 - 你可以无限次地完成交易,但是你每次交易都需要付手续费。如果你已经购买了一个股票,在卖出它之前你就不能再继续购买股票了。
- 返回获得利润的最大值。
示例 1:
输入: prices = [1, 3, 2, 8, 4, 9], fee = 2
输出: 8
解释: 能够达到的最大利润:
在此处买入 prices[0] = 1
在此处卖出 prices[3] = 8
在此处买入 prices[4] = 4
在此处卖出 prices[5] = 9
总利润: ((8 - 1) - 2) + ((9 - 4) - 2) = 8.
注意:
- 0 < prices.length <= 50000.
- 0 < prices[i] < 50000.
- 0 <= fee < 50000.
来源:力扣(LeetCode)第714题
链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-transaction-fee
分析:
- 这道题看似很难,但只要找准状态,就变得很简单。
- 我们把数组长度理解为天数,我们每一天只要知道两个状态。
- cost:手里没有股票的最大利润
- hold:手里有股票的最大利润
- 如果当天是cost,有两种可能,一种是昨天手里就没股票,今天依然不买。第二种昨天手里有股票,今天卖了。
- 如果当天是hold,也有两种可能,一种昨天手里有股票,今天不操作,另一种昨天手里没股票,今天买了。
- 你每次都要选择最优的情况,比如昨天100.今天就10块,如果你手里有股票,那你肯定不会卖。
代码:
class Solution {
public int maxProfit(int[] prices, int fee) {
int cash = 0, hold = -prices[0];
for (int i = 1; i < prices.length; i++) {
cash = Math.max(cash, hold + prices[i] - fee);
hold = Math.max(hold, cash - prices[i]);
}
return cash;
}
}
class Solution:
def maxProfit(self, prices: List[int], fee: int) -> int:
cash, hold = 0, -prices[0]
for i in range(1, len(prices)):
cash = max(cash, prices[i] + hold - fee)
hold = max(hold, cash - prices[i])
return cash
复杂度分析:
- 时间复杂度:O(n)
- 空间复杂度:O(1)