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:
prices = [1, 2, 3, 0, 2]
maxProfit = 3
transactions = [buy, sell, cooldown, buy, sell]
题意:
一个数组,第i个元素是某股票在第i天的价格,设计算法找到最大利润
可以进行多次买卖交易
限制条件是:在下次买入之前要先卖掉,并且卖掉后的第二天要cooldown不可以购买
参考:
http://bookshadow.com/weblog/2015/11/24/leetcode-best-time-to-buy-and-sell-stock-with-cooldown/
引入辅助数组sells
和buys
第i天交易时获得的累计收益只与第i-1天与第i-2天有关
记第i天与第i-1天的价格差:delta = price[i] - price[i - 1]
状态转移方程为:
上述方程的含义为:
而实际上:
所求的最大收益为max(sells)
。显然,卖出股票时才可能获得收益。
Python
class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
size=len(prices)
if not size:
return 0
buys=[None]*size
sells=[None]*size
sells[0]=0
buys[0]=-prices[0]
for x in range(1,size):
delta=prices[x]-prices[x-1]
sells[x]=max(buys[x-1]+prices[x],sells[x-1]+delta)
buys[x]=max(buys[x-1]-delta,sells[x-2]-prices[x] if x>1 else None)
return max(sells)