Say you have an array for which the ith element is the price of a given stock on day i.
Design an algorithm to find the maximum profit. You may complete as many transactions as you like (ie, buy one and sell one share of the stock multiple times) with the following restrictions:
- You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).
- After you sell your stock, you cannot buy stock on next day. (ie, cooldown 1 day)
Example:
Input: [1,2,3,0,2] Output: 3 Explanation: transactions = [buy, sell, cooldown, buy, sell]
LeetCode:链接
本题与LeetCode122:Best Time to Buy and Sell Stock II唯一的区别是在卖出股票后需要间隔至少一天才能再次买入。
一天的状态有:buy
买入,sell
卖出,cooldown
冷却。
但是对于这一天是否持股只有两种状态:持股状态(buy
),没有持股状态(sell,cooldown
)。
对于当天持股状态时,至当天的为止的最大利润有两种可能:1)今天什么也没干,和昨天的持股状态一样;2)今天买了股票,昨天是冷却期,利润是前天卖出股票时候得到的利润减去今天股票的价钱。 二者取最大值。
对于当天未持股状态,至当天为止的最大利润有两种可能:1)今天什么也没干,和昨天的未持股状态一样;2)昨天持有股票,今天卖出了,利润是昨天持有股票时候的利润加上今天股票的价钱。 二者取最大值。
直至最后一天的状态应该是卖出状态。最终利润为sell[-1]。
class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
if not prices:
return 0
n = len(prices)
sell = [0] * n
buy = [0] * n
buy[0] = -prices[0]
for i in range(1, n):
"""
今天未持股
1、今天什么也没干,跟昨天未持股状态一样;
2、昨天持有股票,今天卖出了,利润是昨天持有股票时候的利润加上今天股票的价钱。
二者取最大值。
"""
sell[i] = max(sell[i-1], buy[i-1] + prices[i])
"""
今天持有股票
1、今天什么也没干,跟昨天持股状态一样;
2、今天买入,昨天是冷却期,利润是前天卖出股票时候得到的利润减去今天股票的价钱。
二者取最大值。
"""
buy[i] = max(buy[i-1], (sell[i-2] if i >= 2 else 0) - prices[i])
return sell[-1]