您的位置:首页 > 编程语言 > Python开发

Best Time to Buy and Sell Stock III Leetcode Python

2015-01-26 11:54 447 查看
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 at most two transactions. Note: You may not engage in multiple transactions at the same time (ie, you
must sell the stock before you buy again). 这道题目可以采用两种方法来解,第一种解法是 分别从左往右扫描一次 以及从右往左扫描一次依次把得到的最大的maxpro 放到数组里面 stack1 stack2 最后求解的时候 将stack1[i]+stack2[i+1] 得到的最大值就是最大值。代码如下

class Solution:
# @param prices, a list of integer
# @return an integer
def maxProfit(self, prices):
if len(prices)==0:
return 0
stack1=[]
stack2=[]
lowprice=prices[0]
maxpro=0
highprice=prices[-1]
for index in range(len(prices)):
lowprice=min(lowprice,prices[index])
maxpro=max(maxpro,prices[index]-lowprice)
stack1.append(maxpro)
maxpro=0
for index in reversed(range(len(prices))):
highprice=max(highprice,prices[index])
maxpro=max(maxpro,highprice-prices[index])
stack2.insert(0,maxpro)

for index in range(len(prices)-1):
maxpro=max(maxpro,stack1[index]+stack2[index+1])
return maxpro


第二种做法是用动态规划的方法解的
i表示第i天 j表示一共可以进行j次交易

全局最优的表达式为g[j]=max(g[j],l[j])

局部最优的表达为当前最优和 前一个全局最优加上当前可以挣的钱的最大 表达式为

l[j]=max(g[j-1]+max(dif,0),l[j]+dif)

具体代码如下:

class Solution:
# @param prices, a list of integer
# @return an integer
def maxProfit(self, prices):
g=[0,0,0]
l=[0,0,0]
for i in range(len(prices)-1):
dif=prices[i+1]-prices[i]
for j in reversed(range(1,3)):
l[j]=max(g[j-1]+max(dif,0),l[j]+dif)
g[j]=max(l[j],g[j])
return g[2]
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  python leetcode DP